Searched refs:dims_out (Results 1 – 2 of 2) sorted by relevance
93 std::vector<int>* dims_out = output_array->mutable_shape()->mutable_dims(); in ComputeBinaryOperatorOutputSize() local94 dims_out->clear(); in ComputeBinaryOperatorOutputSize()95 dims_out->reserve(rank_out); in ComputeBinaryOperatorOutputSize()109 dims_out->push_back(dim_y); in ComputeBinaryOperatorOutputSize()113 dims_out->push_back(dim_x); in ComputeBinaryOperatorOutputSize()118 dims_out->push_back(dim_x); in ComputeBinaryOperatorOutputSize()
125 const std::vector<int>& dims_out) { in DescribeSparseTensorDenseMatMul() argument134 DescribeArbitraryRankOutput(dims_out, DT_FLOAT, &op_context.op_info); in DescribeSparseTensorDenseMatMul()