| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.cc | 34 const HloCostAnalysis::ShapeSizeFunction& shape_size) in SimpleCostModel() argument
35 : max_parallelism_(max_parallelism), shape_size_(shape_size) {} in SimpleCostModel()
55 const HloCostAnalysis::ShapeSizeFunction& shape_size, in DefaultCostModel() argument
58 shape_size_(shape_size), in DefaultCostModel()
111 const HloCostAnalysis::ShapeSizeFunction& shape_size, HloModule* module, in ParallelTaskAssignment() argument
116 auto cost_analysis = absl::make_unique<HloCostAnalysis>(shape_size); in ParallelTaskAssignment()
121 cost_model_.reset(new DefaultCostModel(max_parallelism, shape_size, in ParallelTaskAssignment()
127 cost_model_.reset(new SimpleCostModel(max_parallelism, shape_size)); in ParallelTaskAssignment()
|
| D | parallel_task_assignment.h | 42 const HloCostAnalysis::ShapeSizeFunction& shape_size,
69 const HloCostAnalysis::ShapeSizeFunction& shape_size, in ParallelTaskAssigner() argument
72 shape_size_function_(shape_size), in ParallelTaskAssigner()
|
| D | conv_canonicalization_test.cc | 95 [](int64 shape_size) { in TEST_F() argument
157 [](int64 shape_size) { in TEST_F() argument
|
| D | ir_emission_utils_test.cc | 48 [](int64 shape_size) { in TEST_F() argument
|
| D | parallel_task_assignment_test.cc | 38 : HloTestBase(), target_machine_features_([](int64 shape_size) { in ParallelTaskAssignmentTest() argument
|
| D | cpu_layout_assignment_test.cc | 54 [](int64 shape_size) { in AssignLayouts() argument
329 [](int64 shape_size) { in RunDotOutputFusion() argument
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | strided_slice.cc | 93 int shape_size = 0; in CheckOutputSize() local
115 TF_LITE_ENSURE_EQ(context, output_shape->data[shape_size], dim_shape); in CheckOutputSize()
116 shape_size++; in CheckOutputSize()
119 TF_LITE_ENSURE_EQ(context, output_shape->size, shape_size); in CheckOutputSize()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compiler.h | 259 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction() local
260 return [shape_size](const BufferValue& buffer) { in BufferSizeBytesFunction()
261 return shape_size(buffer.shape()); in BufferSizeBytesFunction()
|
| D | hlo_cost_analysis.h | 51 explicit HloCostAnalysis(const ShapeSizeFunction& shape_size);
179 HloCostAnalysis(const ShapeSizeFunction& shape_size,
183 const ShapeSizeFunction& shape_size, const Properties& per_second_rates);
|
| D | hlo_cost_analysis.cc | 40 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size) in HloCostAnalysis() argument
41 : HloCostAnalysis(shape_size, {}) {} in HloCostAnalysis()
43 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size, in HloCostAnalysis() argument
45 : shape_size_(shape_size), per_second_rates_(per_second_rates) {} in HloCostAnalysis()
1029 const ShapeSizeFunction& shape_size, const Properties& per_second_rates) { in CreateNestedCostAnalysis() argument
1030 return absl::WrapUnique(new HloCostAnalysis(shape_size, per_second_rates)); in CreateNestedCostAnalysis()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_util.py | 492 shape_size = nparray.size
495 shape_size = np.prod(shape, dtype=np.int64)
496 is_same_size = shape_size == nparray.size
501 elif nparray.size != shape_size:
510 if nparray.size > shape_size:
513 (shape_size, nparray.size))
519 if is_same_size and numpy_dtype in _TENSOR_CONTENT_TYPES and shape_size > 1:
|
| D | python_op_gen_internal.cc | 415 } else if (value.list().shape_size() > 0) { in AttrListToPython()
416 for (int i = 0; i < value.list().shape_size(); ++i) { in AttrListToPython()
|
| /external/tensorflow/tensorflow/core/example/ |
| D | example_parser_configuration.cc | 74 if (dense_shapes.list().shape_size() != num_dense) { in ExtractExampleParserConfiguration()
76 dense_shapes.list().shape_size(), in ExtractExampleParserConfiguration()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_graph_execution_test.cc | 341 ASSERT_EQ(cni0.shape_size(), cni1.shape_size()); in CompareGraphTransferInfo()
342 for (int j = 0; j < cni0.shape_size(); ++j) { in CompareGraphTransferInfo()
|
| D | graph_transferer_test.cc | 279 ASSERT_EQ(4, params_a->shape_size()); in TEST_F()
288 ASSERT_EQ(4, params_b->shape_size()); in TEST_F()
|
| D | hexagon_control_wrapper.cc | 251 CHECK(params.shape_size() == 4); in SetupGraph()
|
| D | graph_transferer.cc | 1123 CHECK_EQ(params.shape_size(), 4); in DumpNodeTransferParams()
1177 CHECK_EQ(params.shape_size(), 4); in DumpVerificationStringOfNodeTransferParams()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 1025 int64 limit, int shape_size, const T* data, int64* data_index, in PrintOneDim() argument
1030 if (dim_index == shape_size - 1) { in PrintOneDim()
1052 PrintOneDim(dim_index + 1, shape, limit, shape_size, data, data_index, in PrintOneDim()
1141 const int shape_size = tensor_shape.dims(); in SummarizeArray() local
1142 PrintOneDim(0, shape, limit, shape_size, array, &data_index, &ret); in SummarizeArray()
|
| D | attr_value_util.cc | 267 } else if (attr_value.list().shape_size() > 0) { in SummarizeAttrValue()
268 for (int i = 0; i < attr_value.list().shape_size(); ++i) { in SummarizeAttrValue()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 149 RuntimeShape(int shape_size, int32 value) : size_(0) { in RuntimeShape() argument
150 Resize(shape_size); in RuntimeShape()
151 for (int i = 0; i < shape_size; ++i) { in RuntimeShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_util.h | 141 int32* shape_size,
|
| D | llvm_util.cc | 239 int32* shape_size, in EncodeSelfDescribingShapeConstant() argument
245 *shape_size = static_cast<int32>(encoded_shape.size()); in EncodeSelfDescribingShapeConstant()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 258 } else if (attr_value.list().shape_size() > 0) { in PrintAttrValue()
259 for (int i = 0; i < attr_value.list().shape_size(); ++i) { in PrintAttrValue()
284 list.shape_size() == 0 && list.tensor_size() == 0; in IsEmptyList()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 720 int64 shape_size = [&]() { in ValidateShapeSize() local
739 if (shape_size < 0) { in ValidateShapeSize()
744 VLOG(3) << "Shape size is valid: " << shape_size; in ValidateShapeSize()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.h | 337 for (int i = 0; i < attr.second.list().shape_size(); ++i) { in TFGraphNode()
|