Searched refs:CreateConstOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils_test.cc | 31 void CreateConstOp(const string& name, std::initializer_list<int64> dims, in CreateConstOp() function
80 CreateConstOp("input", {batch, rows, cols, in_depth}, input); in TEST()
83 CreateConstOp("filter", {filter_rows, filter_cols, in_depth, out_depth}, in TEST()
87 CreateConstOp("output_backprop", {batch, out_rows, out_cols, out_depth}, in TEST()
|