| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_utils.cc | 132 Status BuildNonNestedTensorList(xla::XlaOp buffer, xla::XlaOp push_index, in BuildNonNestedTensorList() argument
135 *output_list = xla::Tuple(buffer.builder(), {buffer, push_index}); in BuildNonNestedTensorList()
160 Status GetTensorListPushIndex(xla::XlaOp list, xla::XlaOp* push_index) { in GetTensorListPushIndex() argument
168 *push_index = xla::GetTupleElement(list, tuple_size - 1); in GetTensorListPushIndex()
172 Status SetTensorListPushIndex(xla::XlaOp list, xla::XlaOp push_index, in SetTensorListPushIndex() argument
186 result_parts.push_back(push_index); in SetTensorListPushIndex()
366 xla::XlaOp push_index = xla::GetTupleElement(list, list_tuple_size - 1); in ExecuteTensorListPushBack() local
385 start_indices[0] = push_index; in ExecuteTensorListPushBack()
401 start_indices[0] = push_index; in ExecuteTensorListPushBack()
409 xla::XlaOp updated_push_index = push_index + xla::ConstantR0<int32>(b, 1); in ExecuteTensorListPushBack()
[all …]
|
| D | tensor_list_utils.h | 37 Status BuildNonNestedTensorList(xla::XlaOp buffer, xla::XlaOp push_index,
53 Status GetTensorListPushIndex(xla::XlaOp list, xla::XlaOp* push_index);
58 Status SetTensorListPushIndex(xla::XlaOp list, xla::XlaOp push_index,
122 Status ExecuteTensorListFromTensor(int push_index, xla::XlaOp tensor,
|
| D | aggregate_ops.cc | 75 xla::XlaOp push_index; in Compile() local
76 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(ctx->Input(0), &push_index)); in Compile()
77 OP_REQUIRES_OK(ctx, BuildNonNestedTensorList(sum, push_index, &sum)); in Compile()
|
| D | shape_op.cc | 406 xla::XlaOp push_index; in Compile() local
407 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(list, &push_index)); in Compile()
411 SetTensorListPushIndex(new_list, push_index, &result)); in Compile()
|
| /external/cronet/stable/third_party/rust/chromium_crates_io/vendor/clap_builder-4.5.21/src/parser/matches/ |
| D | matched_arg.rs | 74 pub(crate) fn push_index(&mut self, index: usize) { in push_index() method
|
| /external/cronet/tot/third_party/rust/chromium_crates_io/vendor/clap_builder-4.5.21/src/parser/matches/ |
| D | matched_arg.rs | 74 pub(crate) fn push_index(&mut self, index: usize) { in push_index() method
|
| /external/rust/android-crates-io/extra_versions/crates/clap/src/parser/matches/ |
| D | matched_arg.rs | 93 pub(crate) fn push_index(&mut self, index: usize) { in push_index() method
|
| /external/rust/android-crates-io/crates/clap_builder/src/parser/matches/ |
| D | matched_arg.rs | 74 pub(crate) fn push_index(&mut self, index: usize) { in push_index() method
|
| /external/cronet/tot/third_party/rust/chromium_crates_io/vendor/clap_builder-4.5.21/src/parser/ |
| D | arg_matcher.rs | 175 ma.push_index(idx); in add_index_to()
|
| /external/rust/android-crates-io/crates/clap_builder/src/parser/ |
| D | arg_matcher.rs | 175 ma.push_index(idx); in add_index_to()
|
| /external/cronet/stable/third_party/rust/chromium_crates_io/vendor/clap_builder-4.5.21/src/parser/ |
| D | arg_matcher.rs | 175 ma.push_index(idx); in add_index_to()
|
| /external/rust/android-crates-io/extra_versions/crates/clap/src/parser/ |
| D | arg_matcher.rs | 212 ma.push_index(idx); in add_index_to()
|
| /external/mesa3d/src/intel/vulkan/grl/gpu/ |
| D | bvh_rebraid.cl | 152 …varying ushort push_index = subgroup_bit_rank( push_mask ); // lane i gets the index of the i'th s…
154 …varying uint shuffled = intel_sub_group_shuffle( key_value, intel_sub_group_shuffle( push_index, f…
|