Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dscoped_allocator_ops_test.cc127 void BuildNodeDef(const TensorShape& shape, DataType dtype, in BuildNodeDef() function in tensorflow::ScopedAllocatorConcatOpTest
229 BuildNodeDef({3, 4, 4}, DT_HALF, "test", 120, 3); in TEST_F()
261 void BuildNodeDef(const TensorShape& in_shape, DataType dtype, in BuildNodeDef() function in tensorflow::ScopedAllocatorSplitOpTest
280 BuildNodeDef(in_shape, dtype, name, id, num_tensors, out_shapes); in MakeOp()
335 BuildNodeDef({4, 4}, DT_FLOAT, "test", 120, 1, {{4, 4}}); in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device_test.cc117 .BuildNodeDef()); in BM_KernelAndDeviceInit()
140 .BuildNodeDef()); in BM_KernelAndDeviceRun()
Dattr_builder.h107 const NodeDef& BuildNodeDef();
Dattr_builder.cc183 const NodeDef& AttrBuilder::BuildNodeDef() { in BuildNodeDef() function in tensorflow::AttrBuilder
Dexecute.cc226 const auto& node_def = op->MutableAttrs()->BuildNodeDef(); in GetOutputDTypes()
406 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in EagerLocalExecute()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc8857 tensorflow::DeviceType(device_type), builder->BuildNodeDef(), in TF_AttrBuilderCheckCanRunOnDevice()