Searched refs:shapes_slices_val (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 118 Tensor shapes_slices_val(DT_STRING, TensorShape({1})); in TestSinglePartition() local
119 shapes_slices_val.flat<string>()(0) = "4 1 0,4:0,1"; in TestSinglePartition()
120 SetNodeTensorAttr<string>("value", shapes_slices_val, in TestSinglePartition()
329 Tensor shapes_slices_val(DT_STRING, TensorShape({2})); in TestMultiPartition() local
330 shapes_slices_val.flat<string>()(0) = "4 1 0,4:0,1"; in TestMultiPartition()
331 shapes_slices_val.flat<string>()(1) = "4 1 0,4:0,1"; in TestMultiPartition()
332 SetNodeTensorAttr<string>("value", shapes_slices_val, in TestMultiPartition()
|