Home
last modified time | relevance | path

Searched refs:BuildNodeDef (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dscoped_allocator_ops_test.cc128 void BuildNodeDef(const TensorShape& shape, DataType dtype, in BuildNodeDef() function in tensorflow::ScopedAllocatorConcatOpTest
230 BuildNodeDef({3, 4, 4}, DT_HALF, "test", 120, 3); in TEST_F()
262 void BuildNodeDef(const TensorShape& in_shape, DataType dtype, in BuildNodeDef() function in tensorflow::ScopedAllocatorSplitOpTest
281 BuildNodeDef(in_shape, dtype, name, id, num_tensors, out_shapes); in MakeOp()
336 BuildNodeDef({4, 4}, DT_FLOAT, "test", 120, 1, {{4, 4}}); in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dmkl_eager_op_rewrite.cc145 const NodeDef& orig_ndef = orig_op->MutableAttrs()->BuildNodeDef(); in SetupNewOp()
225 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in RewriteConv2D()
233 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in RewriteFusedBatchNormV3()
Dkernel_and_device_test.cc114 .BuildNodeDef()); in BM_KernelAndDeviceInit()
135 .BuildNodeDef()); in BM_KernelAndDeviceRun()
Dattr_builder.h163 const NodeDef& BuildNodeDef();
Dexecute.cc268 const auto& node_def = op->MutableAttrs()->BuildNodeDef(); in GetOutputDTypes()
452 const auto& node_def = op->MutableAttrs()->BuildNodeDef(); in VerifyWrappableInCallOp()
531 const NodeDef& node_def = op->MutableAttrs()->BuildNodeDef(); in ValidateOp()
973 node_def = &op->MutableAttrs()->BuildNodeDef(); in ExtractFunctionInputInfo()
1022 auto ndef = op->MutableAttrs()->BuildNodeDef(); in SetOpDevice()
1082 const NodeDef* node_def = &op->MutableAttrs()->BuildNodeDef(); in GetOrCreateKernelAndDevice()
1186 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in GetOrCreateKernelAndDevice()
1670 op->MutableAttrs()->BuildNodeDef(), op->EagerContext().FuncLibDef(), in EagerRemoteExecute()
Dattr_builder.cc241 const NodeDef& AttrBuilder::BuildNodeDef() { in BuildNodeDef() function in tensorflow::AttrBuilder
/external/tensorflow/tensorflow/core/tfrt/fallback/
Dop_kernel_runner.cc41 StatusOr<tensorflow::NodeDef> BuildNodeDef( in BuildNodeDef() function
116 BuildNodeDef(*op_def, num_args, attr_builder)); in Create()
/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
Dop_handler_selector_test.cc421 s = selector()->SelectFromNodeDef(*op, &op->GetAttrs()->BuildNodeDef(), in TEST_F()
460 s = selector()->SelectFromNodeDef(*op, &op->GetAttrs()->BuildNodeDef(), in TEST_F()
474 s = selector()->SelectFromNodeDef(*op, &op->GetAttrs()->BuildNodeDef(), in TEST_F()
487 s = selector()->SelectFromNodeDef(*op, &op->GetAttrs()->BuildNodeDef(), in TEST_F()
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dop_kernel_runner.cc41 tfrt::StatusOr<tensorflow::NodeDef> BuildNodeDef( in BuildNodeDef() function
96 BuildNodeDef(*op_def, num_args, attr_builder)); in Create()
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dnode_expansion_pass.cc63 const NodeDef& ndef = orig_op->MutableAttrs()->BuildNodeDef(); in Run()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dop_cache.cc43 *op_interface, &op_interface->fallback_attrs_.BuildNodeDef(), in GetOrAddOp()
Dc_api_tfrt.h513 const tensorflow::NodeDef NodeDef() { return fallback_attrs_.BuildNodeDef(); } in NodeDef()
/external/tensorflow/tensorflow/c/experimental/gradients/tape/
Dtape_operation.cc208 forward_op_.attrs.BuildNodeDef(); in Execute()
/external/tensorflow/tensorflow/c/eager/
Dgradients.cc480 forward_op_->attrs.BuildNodeDef(); in Execute()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_copy_node.cc61 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in CreateUncachedKernelAndDeviceOp()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc448 tensorflow::DeviceType(device_type), builder->BuildNodeDef(), in TF_AttrBuilderCheckCanRunOnDevice()