Searched refs:SliceHelper (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op_test.cc | 41 static void SliceHelper(::testing::benchmark::State& state) { in SliceHelper() function
79 SliceHelper<float>(state); in BM_SliceFloat()
85 SliceHelper<std::complex<float>>(state); in BM_SliceComplex64()
91 SliceHelper<bfloat16>(state); in BM_SliceBFloat16()
|
| D | slice_op_test.cc | 40 static void SliceHelper(::testing::benchmark::State& state) { in SliceHelper() function
76 SliceHelper<float>(state); in BM_SliceFloat()
82 SliceHelper<bfloat16>(state); in BM_SliceBFloat16()
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops.cc | 730 class SliceHelper { class
732 explicit SliceHelper(OpKernelContext* ctx) in SliceHelper() function in tensorflow::__anon4a08d0720111::SliceHelper
735 ~SliceHelper() { in ~SliceHelper()
973 SliceHelper<Device, T> slicer(ctx); in Compute()
1238 SliceHelper<Device, T> slicer(ctx); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2125 Status SliceHelper(InferenceContext* c, ShapeHandle begin_value, in SliceHelper() function
2185 SliceHelper<int64>(c, begin_value, sizes_value, &dims)); in SliceShape()
2188 SliceHelper<int32>(c, begin_value, sizes_value, &dims)); in SliceShape()
|