Searched refs:end_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | sample_stats.py | 520 end_shape = [np.prod([x_shape[i] for i in axis])]
521 full_shape = front_shape + end_shape
524 end_shape = [-1]
525 full_shape = array_ops.concat([front_shape, end_shape], axis=0)
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 32 std::initializer_list<int> end_shape, in StridedSliceOpModel() argument
46 BuildInterpreter({input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1682 std::initializer_list<int> end_shape, in StridedSliceOpModel() argument
1690 end_ = AddConstInput(TensorType_INT32, end_data, end_shape); in StridedSliceOpModel()
1698 BuildInterpreter({input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1567 ShapeHandle begin_shape, end_shape, strides_shape; in __anondb9326b22502() local
1569 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &end_shape)); in __anondb9326b22502()
1571 TF_RETURN_IF_ERROR(c->Merge(begin_shape, end_shape, &begin_shape)); in __anondb9326b22502()
|