Searched refs:DescribeArbitraryRankInput (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 88 void DescribeArbitraryRankInput(const std::vector<int>& dims, DataType dtype, in DescribeArbitraryRankInput() function
117 DescribeArbitraryRankInput(dims_a, DT_FLOAT, &op_context.op_info); in DescribeBatchMatMul()
118 DescribeArbitraryRankInput(dims_b, DT_FLOAT, &op_context.op_info); in DescribeBatchMatMul()
130 DescribeArbitraryRankInput({nnz_a, 2}, DT_INT64, &op_context.op_info); in DescribeSparseTensorDenseMatMul()
131 DescribeArbitraryRankInput({nnz_a}, DT_FLOAT, &op_context.op_info); in DescribeSparseTensorDenseMatMul()
132 DescribeArbitraryRankInput({2}, DT_INT64, &op_context.op_info); in DescribeSparseTensorDenseMatMul()
133 DescribeArbitraryRankInput(dims_b, DT_FLOAT, &op_context.op_info); in DescribeSparseTensorDenseMatMul()
385 DescribeArbitraryRankInput({4}, DT_INT32, &op_info); in DescribePoolingOp()
544 DescribeArbitraryRankInput({10000000, 10}, DT_FLOAT, &op_context.op_info); in TEST_F()
545 DescribeArbitraryRankInput({16}, DT_INT64, &op_context.op_info); in TEST_F()
[all …]
|