HomeSort by relevance Sort by last modified time
    Searched full:processing_dims (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
strided_slice_op_impl.h 87 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); local
97 result->bit_casted_shaped<Proxy, NDIM>(processing_dims),
110 result->bit_casted_shaped<Proxy, NDIM>(processing_dims),
123 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); local
137 context->input(4).bit_casted_shaped<Proxy, NDIM>(processing_dims),
146 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); local
158 context->input(4).bit_casted_shaped<Proxy, NDIM>(processing_dims),
171 gtl::InlinedVector<int64, 1> processing_dims(1);
172 processing_dims[0] = 1;
177 result->bit_casted_shaped<Proxy, 1>(processing_dims),
    [all...]
strided_slice_op.cc 130 const int processing_dims = processing_shape.dims(); variable
149 if (processing_dims == NDIM) { \
235 const int processing_dims = processing_shape.dims(); variable
246 if (processing_dims == NDIM) { \
327 const int processing_dims = processing_shape.dims(); variable
334 if (processing_dims == NDIM) { \
352 processing_dims));

Completed in 250 milliseconds