Home
last modified time | relevance | path

Searched refs:CalculateOutputSize (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc207 TEST(UtilsTest, CalculateOutputSize) { in TEST() argument
215 EXPECT_EQ(DataTypeSize(DT_FLOAT) * 4 * 4, CalculateOutputSize(output, 0)); in TEST()
217 CalculateOutputSize(output, 1)); in TEST()
220 EXPECT_EQ(4, CalculateOutputSize(output, -1)); in TEST()
224 EXPECT_EQ(0, CalculateOutputSize(output, 2)); in TEST()
Dop_level_cost_estimator.cc501 const double output_size = CalculateOutputSize(op_info, &unknown_shapes); in PredictOpCountBasedCost()
1079 int64 OpLevelCostEstimator::CalculateOutputSize( in CalculateOutputSize() function in tensorflow::grappler::OpLevelCostEstimator
1266 double output_size = CalculateOutputSize(op_info, &found_unknown_shapes); in PredictSparseTensorDenseMatMul()
1287 result.max_memory = CalculateOutputSize(op_info, &result.inaccurate); in PredictIdentity()
1300 result.persistent_memory = CalculateOutputSize(op_info, &result.inaccurate); in PredictVariable()
1322 costs.max_memory = CalculateOutputSize(op_info, &costs.inaccurate); in PredictMetadata()
1352 const double output_size = CalculateOutputSize(op_info, &unknown_shapes); in PredictGatherOrSlice()
1508 CalculateOutputSize(op_info, &found_unknown_shapes); in PredictMaxPool()
1584 CalculateOutputSize(op_info, &found_unknown_shapes); in PredictAvgPool()
1637 CalculateOutputSize(op_info, &found_unknown_shapes); in PredictAvgPoolGrad()
Dutils.h53 int64 CalculateOutputSize(
Dvirtual_scheduler.cc797 CalculateOutputSize(node_state.output_properties, port_num) * in MarkCurrNodeExecuted()
832 CalculateOutputSize(input_state.output_properties, port) * in MarkCurrNodeExecuted()
913 CalculateOutputSize(node_map_.at(node).output_properties, port); in Summary()
949 CalculateOutputSize(node_map_.at(node).output_properties, port); in Summary()
1047 CalculateOutputSize(nodestate.output_properties, slot); in GenerateRunMetadata()
Dop_level_cost_estimator.h116 int64 CalculateOutputSize(const OpInfo& op_info,
Dutils.cc222 int64 CalculateOutputSize( in CalculateOutputSize() function
Dvirtual_scheduler_test.cc2820 return CalculateOutputSize(output_properties, 0); in TEST_F()