Search Options

Results per page
Sort
Preferred Languages
Advance

Results 91 - 100 of 356 for setAttr (0.14 sec)

  1. tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc

        auto min_consumer = b.getNamedAttr("min_consumer", b.getI32IntegerAttr(0));
        auto bad_consumers =
            b.getNamedAttr("bad_consumers", b.getI32ArrayAttr({1, 2, 3, 4}));
    
        module->setAttr("tf.versions",
                        b.getDictionaryAttr(llvm::ArrayRef<mlir::NamedAttribute>(
                            {producer, min_consumer, bad_consumers})));
    
        return run_status_;
      }
    
      Status run_status_;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 27 08:25:30 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.cc

                continue;
              }
              auto new_attr = mlir::DenseElementsAttr::get(
                  llvm::cast<mlir::ShapedType>(attr.getType()), rand_val);
              inst.setAttr(attr_id, new_attr);
            }
          }
        }
      }
      return module_or;
    }
    
    absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
    GraphdefToSplattedMlirTranslateFunction(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 11:51:44 UTC 2024
    - 14.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

              /*type=*/mlir::TypeAttr::get(value_attr.getType()),
              /*is_mutable=*/nullptr);
          op->setAttr(
              kTfSavedModelExportedNamesAttr,
              builder.getStrArrayAttr(object_names.GetExportedNames(node_id)));
        }
      }
      AdjustBoundInputArgTypes(module);
      module->setAttr("tf_saved_model.semantics", builder.getUnitAttr());
      SortSavedModelModule(module);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

            }
    
            rewriter.setInsertionPoint(quantizing_op);
            Operation* float_op = rewriter.clone(*quantizing_op);
            quantizing_op->setAttr(kDebugModeOpQuantAttrName,
                                   rewriter.getUnitAttr());
            float_op->setAttr(kDebugModeOpFloatAttrName, rewriter.getUnitAttr());
            RewireFloatModelBackbone(quantizing_op, float_op);
            return success();
          }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc

      llvm::SmallVector<mlir::Type, 4> output_types(num_split, output_type);
      *split_op = builder->create<mlir::TF::SplitOp>(
          location, output_types, split_dimension_op.getOutput(), src_input);
      (*split_op)->setAttr(
          kNumSplitAttr,
          builder->getIntegerAttr(builder->getIntegerType(32), num_split));
      return mlir::success();
    }
    
    // Creates a tf::ConcatOp that merges `input` values in `concat_dimension`.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 21:28:13 UTC 2024
    - 34K bytes
    - Viewed (0)
  6. 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)
  7. tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc

      op->replaceAllUsesWith(flex_op);
      op->erase();
      return true;
    }
    
    // Sets the "no_fallback" attribute.
    Value SetNoFallbackAttr(PatternRewriter &rewriter, Value val) {
      val.getDefiningOp()->setAttr(kNoFallbackAttr, rewriter.getUnitAttr());
      return val;
    }
    
    // Returns true if the attr is a float attribute and be equal to value.
    static bool FloatValueEquals(const Attribute &attr, double value) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/resource_device_inference.cc

            if (auto device = result->DeviceForResource(output)) {
              LLVM_DEBUG(llvm::dbgs() << " Setting device = " << *device << "\n");
              identity->setAttr(kDeviceAttr, builder.getStringAttr(*device));
            }
          }
        } else if (auto while_region = dyn_cast<WhileRegionOp>(op)) {
          // For WhileRegion, do local analysis prior to visiting the attached
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 03 03:47:00 UTC 2023
    - 13.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc

        for (int i : llvm::seq<int>(0, metadata.result_devices.size())) {
          func_op.setResultAttr(
              i, kTFDeviceAttr,
              StringAttr::get(context, metadata.result_devices[i]));
        }
    
        func_op->setAttr(kHostAttr, StringAttr::get(context, host));
        func_op.setPublic();
        Block *block = func_op.addEntryBlock();
    
        // Clones and moves the operations into the function's body. And the cloned
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 13.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc

                                         ModuleOp& module) {
      const auto ex_attr_value = module->getAttrOfType<IntegerAttr>(attr_name);
      if (ex_attr_value == nullptr) {
        module->setAttr(attr_name, attr_value);
        return success();
      }
      if (ex_attr_value == attr_value) {
        return success();
      }
      return op->emitOpError() << "module already contains an attribute "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16K bytes
    - Viewed (0)
Back to top