Home
last modified time | relevance | path

Searched refs:CalculateTensorSize (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc181 TEST(UtilsTest, CalculateTensorSize) { in TEST() argument
184 CalculateTensorSize(ShapeToTensorProperty({1}, DT_FLOAT))); in TEST()
186 CalculateTensorSize(ShapeToTensorProperty({4, 4}, DT_FLOAT))); in TEST()
188 CalculateTensorSize(ShapeToTensorProperty({10, 10, 10}, DT_HALF))); in TEST()
191 CalculateTensorSize(ShapeToTensorProperty({100, 7, 8, 99}, DT_FLOAT))); in TEST()
196 EXPECT_EQ(DataTypeSize(DT_FLOAT) * 1, CalculateTensorSize(t)); in TEST()
201 CalculateTensorSize(ShapeToTensorProperty({-1, 7, 8, 99}, DT_FLOAT))); in TEST()
204 CalculateTensorSize(ShapeToTensorProperty({-1, 7, -1, 99}, DT_FLOAT))); in TEST()
Dop_level_cost_estimator.cc1044 int64 OpLevelCostEstimator::CalculateTensorSize( in CalculateTensorSize() function in tensorflow::grappler::OpLevelCostEstimator
1056 int64 input_size = CalculateTensorSize(input, found_unknown_shapes); in CalculateInputSize()
1256 CalculateTensorSize(op_info.inputs(0), &found_unknown_shapes); in PredictSparseTensorDenseMatMul()
1258 CalculateTensorSize(op_info.inputs(1), &found_unknown_shapes); in PredictSparseTensorDenseMatMul()
1260 CalculateTensorSize(op_info.inputs(2), &found_unknown_shapes); in PredictSparseTensorDenseMatMul()
1498 CalculateTensorSize(op_info.inputs(0), &found_unknown_shapes); in PredictMaxPool()
1546 CalculateTensorSize(op_info.inputs(0), &found_unknown_shapes); in PredictMaxPoolGrad()
1548 CalculateTensorSize(op_info.inputs(2), &found_unknown_shapes); in PredictMaxPoolGrad()
1551 CalculateTensorSize(op_info.inputs(0), &found_unknown_shapes); in PredictMaxPoolGrad()
1574 CalculateTensorSize(op_info.inputs(0), &found_unknown_shapes); in PredictAvgPool()
[all …]
Dutils.h49 int64 CalculateTensorSize(const OpInfo::TensorProperties& prop);
Dop_level_cost_estimator.h101 int64 CalculateTensorSize(const OpInfo::TensorProperties& tensor,
Dutils.cc199 int64 CalculateTensorSize(const OpInfo::TensorProperties& prop) { in CalculateTensorSize() function
233 return CalculateTensorSize(output_properties[port_num]); in CalculateOutputSize()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc1157 bytes_to_swap += CalculateTensorSize(t); in SwappingPass()