Searched refs:tensor_description (Results 1 – 21 of 21) sorted by relevance
34 def tensor_description(var): function75 shapes.append(tensor_description(output))106 print(var.name, tensor_description(var), '[%d, bytes: %d]' %
116 output.tensor_description().has_allocation_description()) { in AddMemoryStats()118 int64 output_bytes = std::max(output.tensor_description() in AddMemoryStats()121 output.tensor_description() in AddMemoryStats()125 output.tensor_description().allocation_description().ptr(); in AddMemoryStats()
10 import "tensorflow/core/framework/tensor_description.proto";38 TensorDescription tensor_description = 3; field
867 TensorDescription tensor_description; in AllocatedBytes() local868 FillDescription(&tensor_description); in AllocatedBytes()869 if (tensor_description.has_allocation_description() && in AllocatedBytes()870 tensor_description.allocation_description().allocated_bytes() > 0) { in AllocatedBytes()871 return tensor_description.allocation_description().allocated_bytes(); in AllocatedBytes()
9 import "tensorflow/core/framework/tensor_description.proto";
230 Bytes(output.tensor_description() in BuildCostModel()233 stats.output(i).tensor_description().shape(), in BuildCostModel()236 output.tensor_description() in BuildCostModel()
14 tensorflow/core/framework/tensor_description.pb_text.cc
26 tensorflow/core/framework/tensor_description.proto
21 tensorflow/core/framework/tensor_description.pb.cc
22 tensorflow/core/framework/tensor_description.pb.h
453 desc = str(value.tensor_description).replace('"', '')499 allocation = output.tensor_description.allocation_description
62 const auto& current = output.tensor_description(); in Validate()195 detail->outputs[slot] = output.tensor_description(); in ProcessStepStats()
56 node_stat_dims = node_stat_out.tensor_description.shape.dim
177 live->memory_used = output.tensor_description() in InferFromTrace()
1592 EXPECT_EQ(DT_FLOAT, output.tensor_description().dtype()); in TEST_F()1593 EXPECT_EQ(2, output.tensor_description().shape().dim().size()); in TEST_F()1594 for (const auto& dim : output.tensor_description().shape().dim()) { in TEST_F()
426 const auto size = node_output.tensor_description() in GetStatsStringFromRunMetadata()
119 no.tensor_description().allocation_description().requested_bytes(); in MergeFromStats()
157 "tensorflow/core/framework/tensor_description.proto"
199 tot += no.tensor_description().allocation_description().requested_bytes(); in DetailText()
181 "framework/tensor_description.proto",
1030 output.tensor_description.allocation_description.ptr)