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
917 TensorDescription tensor_description; in AllocatedBytes() local918 FillDescription(&tensor_description); in AllocatedBytes()919 if (tensor_description.has_allocation_description() && in AllocatedBytes()920 tensor_description.allocation_description().allocated_bytes() > 0) { in AllocatedBytes()921 return tensor_description.allocation_description().allocated_bytes(); in AllocatedBytes()
9 import "tensorflow/core/framework/tensor_description.proto";
58 const auto& current = output.tensor_description(); in Validate()195 (*outputs)[slot] = output.tensor_description(); in ProcessStepStats()
358 Bytes(output.tensor_description() in BuildCostModel()361 output.tensor_description().shape(), in BuildCostModel()364 output.tensor_description() in BuildCostModel()
20 tensorflow/core/framework/tensor_description.pb_text.cc
25 tensorflow/core/framework/tensor_description.proto
20 tensorflow/core/framework/tensor_description.pb.cc
20 tensorflow/core/framework/tensor_description.pb.h
453 desc = str(value.tensor_description).replace('"', '')499 allocation = output.tensor_description.allocation_description
56 node_stat_dims = node_stat_out.tensor_description.shape.dim
185 live->memory_used = output.tensor_description() in InferFromTrace()
468 const auto size = node_output.tensor_description() in GetStatsStringFromRunMetadata()
2331 EXPECT_EQ(DT_FLOAT, output.tensor_description().dtype()); in TEST_F()2332 EXPECT_EQ(2, output.tensor_description().shape().dim().size()); in TEST_F()2333 for (const auto& dim : output.tensor_description().shape().dim()) { in TEST_F()
119 no.tensor_description().allocation_description().requested_bytes(); in MergeFromStats()
160 "tensorflow/core/framework/tensor_description.proto"
289 tot += no.tensor_description().allocation_description().requested_bytes(); in DetailText()
208 "framework/tensor_description.proto",
1678 output.tensor_description.allocation_description.ptr)