| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A 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()
|
| H A D | conv_canonicalization_test.cc | 95 [](int64_t shape_size) { in TEST_F()
157 [](int64_t shape_size) { in TEST_F()
|
| H A D | cpu_layout_assignment_test.cc | 54 [](int64_t shape_size) { in AssignLayouts()
328 [](int64_t shape_size) { in RunDotOutputFusion()
|
| H A D | parallel_task_assignment.h | 70 const HloCostAnalysis::ShapeSizeFunction& shape_size, in ParallelTaskAssigner()
|
| H A D | ir_emission_utils_test.cc | 48 [](int64_t shape_size) { in TEST_F()
|
| H A D | parallel_task_assignment_test.cc | 39 : HloTestBase(), target_machine_features_([](int64_t shape_size) { in ParallelTaskAssignmentTest()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A 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
|
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_cost_analysis.h | 62 ShapeSizeFunction shape_size; member
|
| H A D | compiler.h | 362 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction() local
|
| H A D | hlo_cost_analysis.cc | 43 HloCostAnalysis::HloCostAnalysis(ShapeSizeFunction shape_size, in HloCostAnalysis()
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/ |
| D | graph_transfer_info.pb.h | 1924 inline int GraphTransferConstNodeInfo::shape_size() const { in shape_size() function
2220 inline int GraphTransferGraphInputNodeInfo::shape_size() const { in shape_size() function
2336 inline int GraphTransferGraphOutputNodeInfo::shape_size() const { in shape_size() function
|
| D | attr_value.pb.h | 1176 inline int AttrValue_ListValue::shape_size() const { in shape_size() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | runtime_shape.h | 49 RuntimeShape(int shape_size, int32_t value) : size_(0) { in RuntimeShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | llvm_util.cc | 247 int32_t* shape_size, in EncodeSelfDescribingShapeConstant()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | shape_util.cc | 918 int64_t shape_size = [&]() { in ValidateShapeSize() local
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
| D | eager_service.pb.h | 5372 inline int QueueResponse::shape_size() const { in shape_size() function
6408 inline int RunComponentFunctionResponse::shape_size() const { in shape_size() function
|