Home
last modified time | relevance | path

Searched refs:sparse_index (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc114 int64 sparse_index = shape_spec.processing_to_sparse_mapping[i]; in EmitDynamicSlice() local
120 if ((begins_are_dynamic[sparse_index] || in EmitDynamicSlice()
121 ends_are_dynamic[sparse_index]) && in EmitDynamicSlice()
146 int64 sparse_index = shape_spec.processing_to_sparse_mapping[i]; in EmitDynamicSlice() local
162 begin_index = xla::Slice(ctx->Input("begin"), {sparse_index}, in EmitDynamicSlice()
163 {sparse_index + 1}, {1}); in EmitDynamicSlice()
174 end_index = xla::Slice(ctx->Input("end"), {sparse_index}, in EmitDynamicSlice()
175 {sparse_index + 1}, {1}); in EmitDynamicSlice()
298 int64 sparse_index = shape_spec.output_to_sparse_mapping[i]; in Compile() local
300 sparse_index == -1 ? false : ends_are_dynamic[sparse_index]; in Compile()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1715 for (int sparse_index = 0; sparse_index < sparse.dims; ++sparse_index) { in BuildDenseSliceSpec() local
1716 if (ellipsis_seen && IsSet(sparse.new_axis_mask, sparse_index)) in BuildDenseSliceSpec()
1718 if (IsSet(sparse.ellipsis_mask, sparse_index)) ellipsis_seen = true; in BuildDenseSliceSpec()
1722 for (int sparse_index = 0; sparse_index < sparse.dims; ++sparse_index) { in BuildDenseSliceSpec() local
1723 if (IsSet(sparse.new_axis_mask, sparse_index)) continue; in BuildDenseSliceSpec()
1724 if (IsSet(sparse.ellipsis_mask, sparse_index)) { in BuildDenseSliceSpec()
1725 auto next_index = std::min(dense->dims - (sparse.dims - sparse_index) + in BuildDenseSliceSpec()
1742 dense->begin[dense_index] = sparse.begin[sparse_index]; in BuildDenseSliceSpec()
1743 dense->end[dense_index] = sparse.end[sparse_index]; in BuildDenseSliceSpec()
1744 dense->strides[dense_index] = sparse.strides[sparse_index]; in BuildDenseSliceSpec()
[all …]
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc397 int64 sparse_index = in ValidateStridedSliceOp() local
402 shape_spec->output_to_sparse_mapping.push_back(sparse_index); in ValidateStridedSliceOp()