Searched refs:CalculateOutputSize (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | utils_test.cc | 207 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()
|
D | utils.h | 54 int64 CalculateOutputSize(
|
D | virtual_scheduler.cc | 937 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()
|
D | op_level_cost_estimator.cc | 818 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()
|
D | op_level_cost_estimator.h | 279 static int64 CalculateOutputSize(const OpInfo& op_info,
|
D | utils.cc | 223 int64 CalculateOutputSize( in CalculateOutputSize() function
|
D | virtual_scheduler_test.cc | 2997 return CalculateOutputSize(output_properties, 0); in TEST_F()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | ragged_tensor_to_tensor_op.cc | 140 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/ |
D | ragged_tensor_to_tensor_tflite.cc | 249 RuntimeShape CalculateOutputSize( in CalculateOutputSize() function 630 RuntimeShape output_shape = CalculateOutputSize( in Eval()
|