- Sort Score
- Result 10 results
- Languages All
Results 111 - 120 of 242 for get_attr (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
} OpBuilder builder(call); auto new_call = builder.create<CallOp>( call.getLoc(), info.decomposed_callee.getFunctionType().getResults(), new_operands, call->getAttrs()); new_call->setAttr( "f", SymbolRefAttr::get( builder.getContext(), const_cast<func::FuncOp&>(info.decomposed_callee).getName()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
EXPECT_TRUE(isa<TF::XlaCallModuleOp>(lifted_op)); EXPECT_EQ( mlir::cast<StringAttr>(lifted_op->getAttr("_original_entry_function")), "composite_dot_general_fn_1"); EXPECT_EQ( mlir::cast<ArrayAttr>(lifted_dot_general_op->getAttr("precision_config")), builder_.getArrayAttr(SmallVector<Attribute>( 1, mlir::stablehlo::PrecisionAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
devices.push_back(builder.getNamedAttr(name, metadata)); } } op->setAttr(kDevicesAttr, builder.getDictionaryAttr(devices)); } mlir::LogicalResult GetDevicesFromOp(mlir::Operation* op, mlir::TF::RuntimeDevices* devices) { auto devices_attr = op->getAttr(kDevicesAttr); if (!devices_attr) return mlir::success();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
if (auto device = res->getAttrOfType<StringAttr>("device")) { if (!device.getValue().empty()) result_id->setAttr("device", device); else result_id->setAttr("device", compile_device_op); } else if (compile_device_op) { result_id->setAttr("device", compile_device_op); } res.getOutput().replaceAllUsesWith(compile_op->getResult(0)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/custom_op_with_tflite_op.mlir
// tf.MyCustomOp is the result of conversion to a Custom op %2 = "tf.MyCustomOp"(%1, %0) {fused_activation_function = "RELU", int_attr = 2 : i32} : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32> loc("MyCustomOp") %3 = "tfl.exp"(%2) : (tensor<4xf32>) -> tensor<4xf32> loc("exp") func.return %3 : tensor<4xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 14 16:41:28 UTC 2022 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/fake_quant_utils.cc
// TODO(fengliuai): copy the regions. OperationState state(op->getLoc(), op->getName().getStringRef(), parent_op->getOperands(), parent_op->getResultTypes(), op->getAttrs(), op->getSuccessors()); Operation* inlined = builder.create(state); parent_op->replaceAllUsesWith(inlined); parent_op->erase(); } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 4.3K bytes - Viewed (0) -
pkg/registry/resource/resourceclaim/storage/storage.go
TableConvertor: printerstorage.TableConvertor{TableGenerator: printers.NewTableGenerator().With(printersinternal.AddHandlers)}, } options := &generic.StoreOptions{RESTOptions: optsGetter, AttrFunc: resourceclaim.GetAttrs} if err := store.CompleteWithOptions(options); err != nil { return nil, nil, err } statusStore := *store statusStore.UpdateStrategy = resourceclaim.StatusStrategy
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri Nov 18 10:11:16 UTC 2022 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
rewriter.getDenseI64ArrayAttr(kOihwToHwioPermutation)); // [b, 0, 1, f]x[0, 1, i, o]->[b, 0, 1, f] const auto new_dimension_nums = rewriter.getAttr<ConvDimensionNumbersAttr>( /*inputBatchDimension=*/0, /*inputFeatureDimension=*/3, /*inputSpatialDimensions=*/SmallVector<int64_t>{1, 2}, /*kernelInputFeatureDimension=*/2, /*kernelOutputFeatureDimension=*/3,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/c/ops.cc
cc_type v; \ auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); \ Status s = cc_ctx->GetAttr(attr_name, &v); \ Set_TF_Status_from_Status(status, s); \ if (s.ok()) { \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 28 22:41:35 UTC 2022 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
for (const auto& attr_def : op_reg_data->op_def.attr()) { registered_attrs.insert(attr_def.name()); } // Attributes are not in the registered attributes set will be ignored. for (auto& attr : inst->getAttrs()) { if (registered_attrs.find(attr.getName()) == registered_attrs.end()) { attrs_to_ignore->insert( absl::string_view(attr.getName().data(), attr.getName().size())); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0)