Home
last modified time | relevance | path

Searched refs:slice_start (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc56 Eigen::array<Index, 2> slice_start = RandomDims<Index, 2>(0, 250); in TEST() local
61 slice_start[i] = numext::mini(dims[i] - 1, slice_start[i]); in TEST()
62 slice_size[i] = numext::mini(slice_size[i], dims[i] - slice_start[i]); in TEST()
73 data_mapper.getSubMapper(slice_start[0], slice_start[1]), in TEST()
77 slice_dst = src.slice(slice_start, slice_size); in TEST()
/external/grpc-grpc/src/core/lib/slice/
Dpercent_encoding.cc47 const uint8_t* slice_start = GRPC_SLICE_START_PTR(slice); in grpc_percent_encode_slice() local
51 for (p = slice_start; p < slice_end; p++) { in grpc_percent_encode_slice()
63 for (p = slice_start; p < slice_end; p++) { in grpc_percent_encode_slice()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddynamic_stitch_op.cc172 std::vector<int64> slice_start(result_rank); in Compile() local
182 slice_start[0] = src_slice_vector[index_num]; in Compile()
187 xla::Slice(expression, slice_start, slice_limit, stride); in Compile()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dslice_test.cc205 int64 slice_start; member
225 Slice(original, {spec.slice_start}, {spec.slice_limit}, in Run()
230 for (int i = spec.slice_start; i < spec.slice_limit; in Run()
246 return absl::StrFormat("%d_%d_%d_%d", spec.input_dim0, spec.slice_start, in SliceR1TestDataToString()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_constant_folding_test.cc153 const int64 slice_start[] = {4, 2, 3, 1, 5}; in TEST_F() local
163 shape, literal_instruction, slice_start, slice_limits, slice_strides)); in TEST_F()