Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for SetAttr (0.12 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

          for (const auto& attr_name : attr_names) {
            if (!global_iter_id_op->hasAttr(attr_name)) continue;
            offset->setAttr(attr_name, global_iter_id_op->getAttr(attr_name));
            new_global_iter_id->setAttr(attr_name,
                                        global_iter_id_op->getAttr(attr_name));
          }
          // Make the next function to get inlined use a different offset.
          ++offset_value;
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

      // use it in exports or if there are breaking changes
      module->setAttr("tfl.schema_version",
                      builder.getI32IntegerAttr(model->version));
      if (!model->description.empty()) {
        module->setAttr("tfl.description",
                        builder.getStringAttr(model->description));
      }
    
      if (!metadata_attrs.empty()) {
        module->setAttr("tfl.metadata", builder.getDictionaryAttr(metadata_attrs));
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

      Operation* parent_op = op->getParentOp();
      if (!parent_op) return false;
      return HasOutsideCompilationAncestor(parent_op);
    }
    
    Operation* ApplyXlaHostTransferAttr(Operation* op, OpBuilder& builder) {
      op->setAttr("_xla_has_host_transfer", builder.getBoolAttr(true));
      return op;
    }
    
    // Creates a tf._XlaSendFromHost or tf._XlaSendFromHostV2 op. If device ordinal
    // is present, a tf._XlaSendFromHostV2 op is created instead.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

            float_func.emitError("The attr_map attribute is malformed");
            return failure();
          }
          identifier_to_attr.insert(
              {llvm::StringRef(std::string(key_and_value_pair[1])),
               inner_op.getAttr(
                   llvm::StringRef(std::string(key_and_value_pair[1])))});
        }
      }
    
      // Set the attributes for ops with the attr_map attribute.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

      auto new_call = builder.create<CallOpType>(
          call_op.getLoc(),
          lifting_info.lifted_callee.getFunctionType().getResults(), new_operands,
          call_op->getAttrs());
      new_call->setAttr("f",
                        SymbolRefAttr::get(builder.getContext(),
                                           lifting_info.lifted_callee.getName()));
      AddLoadsStoresOutsideControlFlowOp(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

               llvm::all_of(op->getResultTypes(), is_not_variant);
      };
    
      auto is_set_item_legal = [](Operation *op) {
        return op->hasAttr("resize_if_index_out_of_bounds") &&
               op->getAttr("resize_if_index_out_of_bounds")
                   .cast<mlir::BoolAttr>()
                   .getValue();
      };
    
      ConversionTarget target(*context);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
Back to top