Searched refs:GetTensorProto (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 336 void GetTensorProto(const DataType dtype, const std::vector<int64>& shape, in GetTensorProto() function
387 GetTensorProto(DT_INT32, {4}, {x[0], x[1], x[2], x[3]}, in DescribePoolingOp()
953 GetTensorProto(DT_FLOAT, {}, {}, /*tensor_content=*/false, &tensor_proto); in TEST_F()
960 GetTensorProto(DT_INT32, {4}, shape_expected, in TEST_F()
969 GetTensorProto(DT_INT64, {4}, shape_expected, in TEST_F()
978 GetTensorProto(DT_INT32, {4}, shape_expected, in TEST_F()
987 GetTensorProto(DT_INT64, {4}, shape_expected, in TEST_F()
|