Home
last modified time | relevance | path

Searched refs:shape_size (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.cc33 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()
Dparallel_task_assignment.h42 const HloCostAnalysis::ShapeSizeFunction& shape_size,
69 const HloCostAnalysis::ShapeSizeFunction& shape_size, in ParallelTaskAssigner() argument
72 shape_size_function_(shape_size), in ParallelTaskAssigner()
Dconv_canonicalization_test.cc95 [](int64 shape_size) { in TEST_F() argument
157 [](int64 shape_size) { in TEST_F() argument
Dir_emission_utils_test.cc44 [](int64 shape_size) { in TEST() argument
Dparallel_task_assignment_test.cc38 : HloTestBase(), target_machine_features_([](int64 shape_size) { in ParallelTaskAssignmentTest() argument
Dcpu_layout_assignment_test.cc54 [](int64 shape_size) { in AssignLayouts() argument
322 [](int64 shape_size) { in RunDotOutputFusion() argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dcompiler.h251 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()
Dhlo_cost_analysis.h50 explicit HloCostAnalysis(const ShapeSizeFunction& shape_size);
156 HloCostAnalysis(const ShapeSizeFunction& shape_size,
Dhlo_cost_analysis.cc35 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/
Dtensor_util.py508 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:
Dpython_op_gen_internal.cc411 } 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/
Dexample_parser_configuration.cc74 if (dense_shapes.list().shape_size() != num_dense) { in ExtractExampleParserConfiguration()
76 dense_shapes.list().shape_size(), in ExtractExampleParserConfiguration()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_graph_execution_test.cc341 ASSERT_EQ(cni0.shape_size(), cni1.shape_size()); in CompareGraphTransferInfo()
342 for (int j = 0; j < cni0.shape_size(); ++j) { in CompareGraphTransferInfo()
Dgraph_transferer_test.cc279 ASSERT_EQ(4, params_a->shape_size()); in TEST_F()
288 ASSERT_EQ(4, params_b->shape_size()); in TEST_F()
Dhexagon_control_wrapper.cc251 CHECK(params.shape_size() == 4); in SetupGraph()
Dgraph_transferer.cc1123 CHECK_EQ(params.shape_size(), 4); in DumpNodeTransferParams()
1177 CHECK_EQ(params.shape_size(), 4); in DumpVerificationStringOfNodeTransferParams()
/external/tensorflow/tensorflow/core/framework/
Dtensor.cc962 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()
Dattr_value_util.cc244 } 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/
Dtypes.h138 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/
Dtypes.h141 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/
Dllvm_util.h141 int32* shape_size,
Dllvm_util.cc238 int32* shape_size, in EncodeSelfDescribingShapeConstant() argument
244 *shape_size = static_cast<int32>(encoded_shape.size()); in EncodeSelfDescribingShapeConstant()
/external/tensorflow/tensorflow/cc/framework/
Dcc_op_gen.cc253 } 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/
Dshape_util.cc689 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/
Dtfprof_node.h338 for (int i = 0; i < attr.second.list().shape_size(); ++i) { in TFGraphNode()

12