Searched refs:SliceTensorShape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_set.cc | 36 TF_RETURN_IF_ERROR(slice.SliceTensorShape(shape_, &result_shape)); in Register()
82 s = slice.SliceTensorShape(shape_, &target_shape); in Query()
94 s = intersection.SliceTensorShape(shape_, &inter_shape); in Query()
139 s = slice.SliceTensorShape(shape_, &target_shape); in QueryMeta()
151 s = intersection.SliceTensorShape(shape_, &inter_shape); in QueryMeta()
|
| D | tensor_slice_util.h | 155 s = slice_s.SliceTensorShape(shape, &shp_s);
160 s = slice_d.SliceTensorShape(shape, &shp_d);
|
| D | saved_tensor_slice_util.cc | 122 return slice->SliceTensorShape(*shape, shape_slice); in ParseShapeAndSlice()
|
| D | tensor_slice_writer.h | 147 TF_RETURN_IF_ERROR(slice.SliceTensorShape(saved_shape, &sliced_shape)); in Add()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_slice_test.cc | 182 TEST(TensorSliceTest, SliceTensorShape) { in TEST() argument
188 TF_EXPECT_OK(a.SliceTensorShape(x, &y)); in TEST()
201 a.SliceTensorShape(x, &y).ToString()); in TEST()
|
| D | tensor_slice.h | 147 Status SliceTensorShape(const TensorShape& shape,
|
| D | tensor_slice.cc | 239 Status TensorSlice::SliceTensorShape(const TensorShape& shape, in SliceTensorShape() function in tensorflow::TensorSlice
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 207 tensor_slice.SliceTensorShape(tensor_shape, &slice_shape)); in Compute()
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 338 slice_spec.SliceTensorShape(full_tensor_shape, &sliced_shape).IgnoreError(); in IsFullSlice()
|