Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/runtime/NEON/functions/
DNEUnstack.cpp41 inline void setup_slice_coordinates_and_mask(Coordinates &slice_start, int32_t &slice_end_mask, con… in setup_slice_coordinates_and_mask() argument
45 slice_start.set_num_dimensions(input_num_dimensions); in setup_slice_coordinates_and_mask()
49 slice_start.set(k, 0); in setup_slice_coordinates_and_mask()
79 Coordinates slice_start; in configure() local
81 …setup_slice_coordinates_and_mask(slice_start, slice_end_mask, input->info()->tensor_shape().num_di… in configure()
85 slice_start.set(axis_u, slice); in configure()
86 …_strided_slice_vector[slice].configure(input, output_vector[slice], slice_start, Coordinates(), Bi… in configure()
101 Coordinates slice_start; in validate() local
105 slice_start.set(wrap_axis(axis, input), k); in validate()
106 …setup_slice_coordinates_and_mask(slice_start, slice_end_mask, input->tensor_shape().num_dimensions… in validate()
[all …]
/external/ComputeLibrary/src/runtime/CL/functions/
DCLUnstack.cpp41 inline void setup_slice_coordinates_and_mask(Coordinates &slice_start, int32_t &slice_end_mask, con… in setup_slice_coordinates_and_mask() argument
45 slice_start.set_num_dimensions(input_num_dimensions); in setup_slice_coordinates_and_mask()
49 slice_start.set(k, 0); in setup_slice_coordinates_and_mask()
84 Coordinates slice_start; in configure() local
86 …setup_slice_coordinates_and_mask(slice_start, slice_end_mask, input->info()->tensor_shape().num_di… in configure()
90 slice_start.set(axis_u, slice); in configure()
91 …_vector[slice].configure(compile_context, input, output_vector[slice], slice_start, Coordinates(),… in configure()
104 Coordinates slice_start; in validate() local
108 slice_start.set(wrap_axis(axis, input), k); in validate()
109 …setup_slice_coordinates_and_mask(slice_start, slice_end_mask, input->tensor_shape().num_dimensions… in validate()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc63 Eigen::array<Index, 2> slice_start = RandomDims<Index, 2>(0, 250); in TEST() local
68 slice_start[i] = numext::mini(dims[i] - 1, slice_start[i]); in TEST()
69 slice_size[i] = numext::mini(slice_size[i], dims[i] - slice_start[i]); in TEST()
80 data_mapper.getSubMapper(slice_start[0], slice_start[1]), in TEST()
84 slice_dst = src.slice(slice_start, slice_size); in TEST()
/external/rust/crates/grpcio-sys/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/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/eigen/unsupported/test/
Dcxx11_tensor_block_eval.cpp468 DSizes<Index, NumDims> slice_start = RandomDims<NumDims>(5, 10); in test_eval_tensor_slice() local
473 slice_start[i] = numext::mini(dims[i] - 1, slice_start[i]); in test_eval_tensor_slice()
474 slice_size[i] = numext::mini(slice_size[i], dims[i] - slice_start[i]); in test_eval_tensor_slice()
478 input.slice(slice_start, slice_size), in test_eval_tensor_slice()
482 input.slice(slice_start, slice_size), in test_eval_tensor_slice()
722 DSizes<Index, NumDims> slice_start = RandomDims<NumDims>(5, 10); in test_assign_to_tensor_slice() local
727 slice_start[i] = numext::mini(dims[i] - 1, slice_start[i]); in test_assign_to_tensor_slice()
728 slice_size[i] = numext::mini(slice_size[i], dims[i] - slice_start[i]); in test_assign_to_tensor_slice()
734 tensor, map.slice(slice_start, slice_size), in test_assign_to_tensor_slice()
738 tensor, map.slice(slice_start, slice_size), in test_assign_to_tensor_slice()
[all …]
Dcxx11_tensor_executor.cpp365 auto slice_start = DSizes<Index, NumDims>(RandomDims<NumDims>()); in test_execute_slice_rvalue() local
370 slice_start[i] = numext::mini(dims[i] - 1, slice_start[i]); in test_execute_slice_rvalue()
371 slice_size[i] = numext::mini(slice_size[i], dims[i] - slice_start[i]); in test_execute_slice_rvalue()
375 src.slice(slice_start, slice_size); in test_execute_slice_rvalue()
380 auto expr = src.slice(slice_start, slice_size); in test_execute_slice_rvalue()
405 auto slice_start = DSizes<Index, NumDims>(RandomDims<NumDims>(1, 10)); in test_execute_slice_lvalue() local
410 slice_start[i] = numext::mini(dims[i] - 1, slice_start[i]); in test_execute_slice_lvalue()
411 slice_size[i] = numext::mini(slice_size[i], dims[i] - slice_start[i]); in test_execute_slice_lvalue()
419 golden.slice(slice_start, slice_size) = slice; in test_execute_slice_lvalue()
423 auto expr = dst.slice(slice_start, slice_size); in test_execute_slice_lvalue()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddynamic_stitch_op.cc182 std::vector<int64> slice_start(result_rank); in Compile() local
192 slice_start[0] = src_slice_vector[index_num]; in Compile()
197 xla::Slice(expression, slice_start, slice_limit, stride); in Compile()
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py448 slice_start = slice_spec.start
449 if slice_start is None:
450 slice_start = 0 if slice_step > 0 else self._shape[0] - 1
451 elif slice_start < 0:
452 slice_start += self._shape[0]
469 cur = slice_start
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion.cc411 int64_t slice_start = 0; in CreateFusedComputation() local
417 int64_t slice_limit = slice_start + ShapeUtil::ElementsIn(shape); in CreateFusedComputation()
420 MakeSliceHlo(concated_output, {slice_start}, {slice_limit}, in CreateFusedComputation()
422 slice_start = slice_limit; in CreateFusedComputation()
/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()
Dvalue_inference_test.cc237 auto slice_start = ConstantR0(&b, 1); in TEST_F() local
238 auto dynamic_slice = DynamicSlice(constant, {slice_start}, {1}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_constant_folding_test.cc155 const int64 slice_start[] = {4, 2, 3, 1, 5}; in TEST_F() local
165 shape, literal_instruction, slice_start, slice_limits, slice_strides)); in TEST_F()
/external/tensorflow/tensorflow/python/keras/layers/
Dtensorflow_op_layer_test.py583 slice_start = keras.Input(shape=(), dtype='int32')
587 out = inp[..., first_dim[0], slice_start[0]:slice_stop[0]:slice_stride[0]]
589 inputs=[inp, first_dim, slice_start, slice_stop, slice_stride],
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc294 llvm::SmallVector<int64_t, 8> slice_start(slice_end.size(), 0); in matchAndRewrite() local
299 slice_start[map_index] += offset; in matchAndRewrite()
306 slice_shape[i] = slice_end[i] - slice_start[i]; in matchAndRewrite()
312 GetI64ElementsAttr(slice_start, &rewriter), in matchAndRewrite()
3482 auto slice_start = (*(start.begin() + dimension)).getSExtValue(); in matchAndRewrite() local
3503 if (running_size + dim_size > slice_start && in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc737 Value slice_start = CreateI32SplatConst(loc, rewriter, {1}, 1); in CreateCondTrueBranch() local
739 loc, RankedTensorType::get({-1}, shape_dtype), input_shape, slice_start, in CreateCondTrueBranch()
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_510.h87649 long unsigned int slice_start[2]; member
Dvmlinux.h87649 long unsigned int slice_start[2]; member
/external/libbpf/.github/actions/build-selftests/
Dvmlinux.h56918 long unsigned int slice_start[2]; member
/external/bcc/libbpf-tools/arm64/
Dvmlinux.h54376 long unsigned int slice_start[2]; member
Dvmlinux_510.h54376 long unsigned int slice_start[2]; member