Home
last modified time | relevance | path

Searched refs:BuildNodeDef (Results 1 – 11 of 11) 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.cc150 const NodeDef& orig_ndef = orig_op->MutableAttrs()->BuildNodeDef(); in SetupNewOp()
246 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in RewriteConv2D()
254 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();
Dattr_builder.cc224 const NodeDef& AttrBuilder::BuildNodeDef() { in BuildNodeDef() function in tensorflow::AttrBuilder
Dexecute.cc240 const auto& node_def = op->MutableAttrs()->BuildNodeDef(); in GetOutputDTypes()
487 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in GetOrCreateKernelAndDevice()
962 op->MutableAttrs()->BuildNodeDef(), op->EagerContext().FuncLibDef(), in EagerRemoteExecute()
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dnode_expansion_pass.cc63 const NodeDef& ndef = orig_op->MutableAttrs()->BuildNodeDef(); in Run()
/external/tensorflow/tensorflow/c/experimental/gradients/tape/
Dtape_operation.cc208 forward_op_.attrs.BuildNodeDef(); in Execute()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_copy_node.cc60 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in CreateUncachedKernelAndDeviceOp()
/external/tensorflow/tensorflow/c/eager/
Dgradients.cc479 forward_op_->attrs.BuildNodeDef(); in Execute()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc439 tensorflow::DeviceType(device_type), builder->BuildNodeDef(), in TF_AttrBuilderCheckCanRunOnDevice()