Searched refs:shape_slice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 77 shape_slice = array_ops.slice(b1_shape, [0],
80 [shape_slice, [b1_order], [1], [b1_order], [1]], 0)
82 [shape_slice, [1], [b2_order], [1], [b2_order]], 0)
88 kprod_shape = array_ops.concat([shape_slice, [order_prod], [order_prod]], 0)
111 shape_slice = array_ops.slice(shape, [0], shape_slice_size)
112 shape_vec_da = array_ops.concat([shape_slice, [order * order], [1]], 0)
|
| /external/tensorflow/tensorflow/core/util/ |
| D | saved_tensor_slice_util.cc | 91 TensorSlice* slice, TensorShape* shape_slice) { in ParseShapeAndSlice() argument
122 return slice->SliceTensorShape(*shape, shape_slice); in ParseShapeAndSlice()
|
| D | saved_tensor_slice_util.h | 57 TensorSlice* slice, TensorShape* shape_slice);
|