/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | parallel_task_assignment.cc | 33 const HloCostAnalysis::ShapeSizeFunction& shape_size) in SimpleCostModel() argument 34 : max_parallelism_(max_parallelism), shape_size_(shape_size) {} in SimpleCostModel() 54 const HloCostAnalysis::ShapeSizeFunction& shape_size, in DefaultCostModel() argument 57 shape_size_(shape_size), in DefaultCostModel() 109 const HloCostAnalysis::ShapeSizeFunction& shape_size, HloModule* module, in ParallelTaskAssignment() argument 114 auto cost_analysis = absl::make_unique<HloCostAnalysis>(shape_size); in ParallelTaskAssignment() 119 cost_model_.reset(new DefaultCostModel(max_parallelism, shape_size, in ParallelTaskAssignment() 125 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 | 44 [](int64 shape_size) { in TEST() 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 322 [](int64 shape_size) { in RunDotOutputFusion() argument
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | compiler.h | 251 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction() local 252 return [shape_size](const BufferValue& buffer) { in BufferSizeBytesFunction() 253 return shape_size(buffer.shape()); in BufferSizeBytesFunction()
|
D | hlo_cost_analysis.h | 50 explicit HloCostAnalysis(const ShapeSizeFunction& shape_size); 156 HloCostAnalysis(const ShapeSizeFunction& shape_size,
|
D | hlo_cost_analysis.cc | 35 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size) in HloCostAnalysis() argument 36 : HloCostAnalysis(shape_size, {}) {} in HloCostAnalysis() 38 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size, in HloCostAnalysis() argument 40 : shape_size_(shape_size), per_second_rates_(per_second_rates) {} in HloCostAnalysis()
|
/external/tensorflow/tensorflow/python/framework/ |
D | tensor_util.py | 508 shape_size = nparray.size 511 shape_size = np.prod(shape, dtype=np.int64) 512 is_same_size = shape_size == nparray.size 517 elif nparray.size != shape_size: 526 if nparray.size > shape_size: 529 (shape_size, nparray.size)) 535 if is_same_size and numpy_dtype in _TENSOR_CONTENT_TYPES and shape_size > 1:
|
D | python_op_gen_internal.cc | 411 } else if (value.list().shape_size() > 0) { in AttrListToPython() 412 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 | 962 int64 limit, int shape_size, const T* data, int64* data_index, in PrintOneDim() argument 967 if (dim_index == shape_size - 1) { in PrintOneDim() 989 PrintOneDim(dim_index + 1, shape, limit, shape_size, data, data_index, in PrintOneDim() 1078 const int shape_size = tensor_shape.dims(); in SummarizeArray() local 1079 PrintOneDim(0, shape, limit, shape_size, array, &data_index, &ret); in SummarizeArray()
|
D | attr_value_util.cc | 244 } else if (attr_value.list().shape_size() > 0) { in SummarizeAttrValue() 245 for (int i = 0; i < attr_value.list().shape_size(); ++i) { in SummarizeAttrValue()
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
D | types.h | 138 RuntimeShape(int shape_size, int32 value) : size_(0) { in RuntimeShape() argument 139 Resize(shape_size); in RuntimeShape() 140 for (int i = 0; i < shape_size; ++i) { in RuntimeShape()
|
/external/tensorflow/tensorflow/lite/kernels/internal/ |
D | types.h | 141 RuntimeShape(int shape_size, int32 value) : size_(0) { in RuntimeShape() argument 142 Resize(shape_size); in RuntimeShape() 143 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 | 238 int32* shape_size, in EncodeSelfDescribingShapeConstant() argument 244 *shape_size = static_cast<int32>(encoded_shape.size()); in EncodeSelfDescribingShapeConstant()
|
/external/tensorflow/tensorflow/cc/framework/ |
D | cc_op_gen.cc | 253 } else if (attr_value.list().shape_size() > 0) { in PrintAttrValue() 254 for (int i = 0; i < attr_value.list().shape_size(); ++i) { in PrintAttrValue() 279 list.shape_size() == 0 && list.tensor_size() == 0; in IsEmptyList()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | shape_util.cc | 689 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 | 338 for (int i = 0; i < attr.second.list().shape_size(); ++i) { in TFGraphNode()
|