D | scoped_allocator_ops_test.cc | 127 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()
|