- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 97 for get_attr (0.23 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc
return failure(); } res.push_back(signed_lhs_val / signed_rhs_val); } auto res_attr = DenseElementsAttr::get( const_oprs[0].getType().cast<RankedTensorType>(), res); rewriter.replaceOpWithNewOp<stablehlo::ConstantOp>(adaptor.value().Op(), res_attr); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 06:11:55 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_to_legacy_compile_and_replicate_attributes.cc
if (failed(TF::HasValidCompilationAndReplicationAttributes(*op))) return WalkResult::interrupt(); if (op->hasAttr(TF::kReplicationInfoAttr)) { op->setAttr(TF::kTpuReplicateAttr, op->getAttr(TF::kReplicationInfoAttr)); op->removeAttr(TF::kReplicationInfoAttr); op->removeAttr(TF::kCompileDeviceTypeAttr); } return mlir::WalkResult::advance(); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_ops.cc
<< type_string() << " on an XLA device. This should never happen."; } XlaAssignVariableOp::XlaAssignVariableOp(OpKernelConstruction* c) : OpKernel(c) { OP_REQUIRES_OK(c, c->GetAttr("dtype", &dtype_)); } void XlaAssignVariableOp::Compute(OpKernelContext* context) { OP_REQUIRES(context, dtype_ == context->input(1).dtype(), errors::InvalidArgument(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/infeed_ops_xla_adjust_layout.cc
SmallVector<Type> result_types(op.getResultTypes().begin(), op.getResultTypes().end()); if (!op->getAttr("layout")) { auto layout = mlir::GetTPUInfeedLayout(result_types, builder); if (failed(layout)) return; op->setAttr("layout", layout.value()); } } }; void InfeedsOpsXlaAdjustLayout::runOnOperation() { getOperation().walk(runOnInfeedOp); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
} } if (!in_order) { // Functions do not get reused in practice, so skip the check for if the // callee has been updated. StringAttr callee_sym = cluster_func_op.getFuncAttr().getAttr(); MoveResourceArgsToEnd(symtab.lookup<func::FuncOp>(callee_sym)); } builder.setInsertionPoint(cluster_func_op); auto xla_launch_op = builder.create<TF::XlaLaunchOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
pkg/registry/resource/resourceslice/storage/storage.go
TableConvertor: printerstorage.TableConvertor{TableGenerator: printers.NewTableGenerator().With(printersinternal.AddHandlers)}, } options := &generic.StoreOptions{ RESTOptions: optsGetter, AttrFunc: resourceslice.GetAttrs, TriggerFunc: resourceslice.TriggerFunc, Indexers: resourceslice.Indexers(), } if err := store.CompleteWithOptions(options); err != nil { return nil, err } return &REST{store}, nil
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Thu Mar 07 21:22:55 UTC 2024 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_variables.cc
void runOnOperation() override { auto module = getOperation(); // If TFLite variable legalization is not allowed, then we skip this pass. if (auto legalize_tfl_variables_attr = module->getAttr(kLegalizeTflVariables)) { if (!mlir::cast<BoolAttr>(legalize_tfl_variables_attr).getValue()) return; } RewritePatternSet patterns(&getContext()); populateWithGenerated(patterns);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
pkg/registry/core/configmap/storage/storage.go
TableConvertor: printerstorage.TableConvertor{TableGenerator: printers.NewTableGenerator().With(printersinternal.AddHandlers)}, } options := &generic.StoreOptions{ RESTOptions: optsGetter, AttrFunc: configmap.GetAttrs, } if err := store.CompleteWithOptions(options); err != nil { return nil, err } return &REST{store}, nil } // Implement ShortNamesProvider var _ rest.ShortNamesProvider = &REST{}
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Feb 27 20:38:11 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize_compile_and_replicate_attributes.cc
// Convert `_tpu_replicate`. if (op->hasAttr(TF::kTpuReplicateAttr)) { op->setAttr(tensorflow::kReplicationInfoAttr, op->getAttr(TF::kTpuReplicateAttr)); op->removeAttr(tensorflow::kTpuReplicateAttr); op->setAttr(tensorflow::kCompileDeviceTypeAttr, builder.getStringAttr(tensorflow::kTpuDevice)); } // Convert `_XlaMustCompile`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.5K bytes - Viewed (0) -
pkg/registry/core/secret/storage/storage.go
TableConvertor: printerstorage.TableConvertor{TableGenerator: printers.NewTableGenerator().With(printersinternal.AddHandlers)}, } options := &generic.StoreOptions{ RESTOptions: optsGetter, AttrFunc: secret.GetAttrs, } if err := store.CompleteWithOptions(options); err != nil { return nil, err } return &REST{store}, nil
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Feb 27 20:38:11 UTC 2023 - 2K bytes - Viewed (0)