Home
last modified time | relevance | path

Searched refs:CalculateOutputSize (Results 1 – 9 of 9) 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()
Dutils.h54 int64 CalculateOutputSize(
Dvirtual_scheduler.cc937 CalculateOutputSize(node_state.output_properties, port_num) * in MarkNodeExecuted()
997 CalculateOutputSize(input_state.output_properties, port) * in MarkNodeExecuted()
1068 CalculateOutputSize(node_map_.at(node).output_properties, port); in Summary()
1120 CalculateOutputSize(node_map_.at(node).output_properties, port); in Summary()
1223 CalculateOutputSize(nodestate.output_properties, slot); in GenerateRunMetadata()
1299 CalculateOutputSize(node_map_.at(node).output_properties, port); in GetPersistentMemoryUsage()
Dop_level_cost_estimator.cc818 const double output_size = CalculateOutputSize(op_info, &unknown_shapes); in PredictOpCountBasedCost()
1587 int64 OpLevelCostEstimator::CalculateOutputSize(const OpInfo& op_info, in CalculateOutputSize() function in tensorflow::grappler::OpLevelCostEstimator
1854 int64 output_size = CalculateOutputSize(op_info, &found_unknown_shapes); in PredictSparseTensorDenseMatMul()
1897 node_costs->max_memory = CalculateOutputSize(op_info, &inaccurate); in PredictIdentity()
1916 node_costs->persistent_memory = CalculateOutputSize(op_info, &inaccurate); in PredictVariable()
1942 node_costs->max_memory = CalculateOutputSize(op_info, &inaccurate); in PredictMetadata()
1971 const int64 output_size = CalculateOutputSize(op_info, &unknown_shapes); in PredictGatherOrSlice()
2194 const int64 output_size = CalculateOutputSize(op_info, &found_unknown_shapes); in PredictMaxPool()
2311 const int64 output_size = CalculateOutputSize(op_info, &found_unknown_shapes); in PredictAvgPool()
Dop_level_cost_estimator.h279 static int64 CalculateOutputSize(const OpInfo& op_info,
Dutils.cc223 int64 CalculateOutputSize( in CalculateOutputSize() function
Dvirtual_scheduler_test.cc2997 return CalculateOutputSize(output_properties, 0); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_tensor_op.cc140 Status CalculateOutputSize(INDEX_TYPE first_dim, OpKernelContext* c, in CalculateOutputSize() function in tensorflow::__anonf73c43690111::RaggedTensorToTensorBaseOp
351 CalculateOutputSize(first_dimension, context, &output_size)); in Compute()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc249 RuntimeShape CalculateOutputSize( in CalculateOutputSize() function
630 RuntimeShape output_shape = CalculateOutputSize( in Eval()