Searched defs:shape_size (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.cc | 35 const HloCostAnalysis::ShapeSizeFunction& shape_size) in SimpleCostModel()
57 const HloCostAnalysis::ShapeSizeFunction& shape_size, in DefaultCostModel()
114 const HloCostAnalysis::ShapeSizeFunction& shape_size, HloModule* module, in ParallelTaskAssignment()
|
| D | conv_canonicalization_test.cc | 95 [](int64_t shape_size) { in TEST_F()
157 [](int64_t shape_size) { in TEST_F()
|
| D | ir_emission_utils_test.cc | 48 [](int64_t shape_size) { in TEST_F()
|
| D | parallel_task_assignment.h | 70 const HloCostAnalysis::ShapeSizeFunction& shape_size, in ParallelTaskAssigner()
|
| D | parallel_task_assignment_test.cc | 39 : HloTestBase(), target_machine_features_([](int64_t shape_size) { in ParallelTaskAssignmentTest()
|
| D | cpu_layout_assignment_test.cc | 54 [](int64_t shape_size) { in AssignLayouts()
328 [](int64_t shape_size) { in RunDotOutputFusion()
|
| /external/pytorch/torch/csrc/jit/passes/onnx/ |
| D | constant_map.cpp | 112 auto shape_size = ConstantValueMap::GetShape(value_name).value(); in GetShapeInto1DInt64Vector() local
125 auto shape_size = ConstantValueMap::GetShape(value_name).value(); in GetShapeInto1DInt64VectorWithOneUnknown() local
|
| D | shape_type_inference.cpp | 508 int shape_size = static_cast<int>(shape_vector.size()); in ComputeShapeFromReshape() local
634 const std::vector<int64_t>& shape_size) { in UpdateShapeFromVector()
674 std::vector<int64_t> shape_size; in GetValueFromListConstructNode() local
691 std::vector<c10::ShapeSymbol> shape_size; in SetShapeValueFromListConstructNode() local
841 std::vector<c10::ShapeSymbol> shape_size; in ProcessShapeValueForConcatNode() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.h | 62 ShapeSizeFunction shape_size; member
|
| D | compiler.h | 362 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction() local
|
| D | hlo_cost_analysis.cc | 43 HloCostAnalysis::HloCostAnalysis(ShapeSizeFunction shape_size, in HloCostAnalysis()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | runtime_shape.h | 49 RuntimeShape(int shape_size, int32_t value) : size_(0) { in RuntimeShape()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 1090 int64_t limit, int shape_size, const T* data, in PrintOneDim()
1206 const int shape_size = tensor_shape.dims(); in SummarizeArray() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_util.cc | 247 int32_t* shape_size, in EncodeSelfDescribingShapeConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 918 int64_t shape_size = [&]() { in ValidateShapeSize() local
|