Searched refs:DescribeArbitraryRankOutput (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | op_level_cost_estimator_test.cc | 100 void DescribeArbitraryRankOutput(const std::vector<int>& dims, DataType dtype, in DescribeArbitraryRankOutput() function 134 DescribeArbitraryRankOutput(dims_out, DT_FLOAT, &op_context.op_info); in DescribeSparseTensorDenseMatMul() 546 DescribeArbitraryRankOutput({16, 10}, DT_FLOAT, &op_context.op_info); in TEST_F() 584 DescribeArbitraryRankOutput({10, 10}, DT_FLOAT, &op_context.op_info); in TEST_F()
|