Searched defs:shape_size (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.cc | 33 const HloCostAnalysis::ShapeSizeFunction& shape_size) in SimpleCostModel()
54 const HloCostAnalysis::ShapeSizeFunction& shape_size, in DefaultCostModel()
109 const HloCostAnalysis::ShapeSizeFunction& shape_size, HloModule* module, in ParallelTaskAssignment()
|
| D | conv_canonicalization_test.cc | 95 [](int64 shape_size) { in TEST_F()
157 [](int64 shape_size) { in TEST_F()
|
| D | ir_emission_utils_test.cc | 44 [](int64 shape_size) { in TEST()
|
| D | parallel_task_assignment.h | 69 const HloCostAnalysis::ShapeSizeFunction& shape_size, in ParallelTaskAssigner()
|
| D | parallel_task_assignment_test.cc | 38 : HloTestBase(), target_machine_features_([](int64 shape_size) { in ParallelTaskAssignmentTest()
|
| D | cpu_layout_assignment_test.cc | 54 [](int64 shape_size) { in AssignLayouts()
322 [](int64 shape_size) { in RunDotOutputFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compiler.h | 251 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction() local
|
| D | hlo_cost_analysis.cc | 35 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size) in HloCostAnalysis()
38 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size, in HloCostAnalysis()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 962 int64 limit, int shape_size, const T* data, int64* data_index, in PrintOneDim()
1078 const int shape_size = tensor_shape.dims(); in SummarizeArray() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_util.cc | 238 int32* shape_size, in EncodeSelfDescribingShapeConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 689 int64 shape_size = [&]() { in ValidateShapeSize() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 138 RuntimeShape(int shape_size, int32 value) : size_(0) { in RuntimeShape()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 141 RuntimeShape(int shape_size, int32 value) : size_(0) { in RuntimeShape()
|