| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| scatter_test.cc | 31 Literal* scatter_indices, Literal* updates) {
32 RunTest(hlo_text, {operand, scatter_indices, updates});
67 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); local
69 RunTest(hlo_text, &operand, &scatter_indices, &updates);
98 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 1}); local
100 RunTest(hlo_text, &operand, &scatter_indices, &updates);
126 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); local
129 RunTest(hlo_text, &operand, &scatter_indices, &updates);
195 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0, 0}}); local
196 RunTest(hlo_text, &operand, &scatter_indices, &updates)
223 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); local
252 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); local
281 Literal scatter_indices = LiteralUtil::CreateR1<int32>({2, 1}); local
311 Literal scatter_indices = LiteralUtil::CreateR1<int32>({1, 1}); local
340 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 2}, {2, 1}}); local
371 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); local
401 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); local
429 Literal scatter_indices = LiteralUtil::CreateR1<int32>({1, 1}); local
457 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{2, 1}, {1, 1}}); local
484 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); local
512 Literal scatter_indices = local
541 Literal scatter_indices = LiteralUtil::CreateR2<int32>( local
571 Literal scatter_indices = LiteralUtil::CreateR2<uint32>( local
601 Literal scatter_indices = LiteralUtil::CreateR2<int32>( local
633 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 2}}); local
661 Literal scatter_indices = LiteralUtil::CreateR0<int32>(1); local
689 Literal scatter_indices = LiteralUtil::CreateR0<int32>(1); local
716 Literal scatter_indices = LiteralUtil::CreateR1<int32>({}); local
743 Literal scatter_indices = LiteralUtil::CreateR1<int32>({}); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| scatter_expander.cc | 29 // Transposes the given scatter_indices such that the index_vector_dim becomes
32 HloInstruction* scatter_indices, int64 index_vector_dim) {
33 const Shape& scatter_indices_shape = scatter_indices->shape();
36 return scatter_indices;
40 return scatter_indices;
51 return MakeTransposeHlo(scatter_indices, permutation);
54 // Canonicalizes the scatter_indices tensor in order to keep them uniform while
57 HloInstruction* scatter_indices, int64 index_vector_dim) {
61 TransposeIndexVectorDimToLast(scatter_indices, index_vector_dim));
62 if (scatter_indices->shape().rank() == index_vector_dim + 1 &
226 HloInstruction* scatter_indices = loop_state[1]; local
348 HloInstruction* scatter_indices = scatter->mutable_operand(1); local
[all...] |
| hlo_evaluator_test.cc | 2335 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); local
2368 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); local
2403 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); local
2437 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); local
2471 Literal scatter_indices = LiteralUtil::CreateR1<int32>({2, 1}); local
2507 Literal scatter_indices = LiteralUtil::CreateR1<int32>({1, 1}); local
2541 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 2}, {2, 1}}); local
2577 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); local
2615 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); local
2650 Literal scatter_indices = LiteralUtil::CreateR1<int32>({1, 1}); local
2683 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{2, 1}, {1, 1}}); local
2715 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); local
2747 Literal scatter_indices = local
2783 Literal scatter_indices = LiteralUtil::CreateR1<int32>({-1, 2}); local
2817 Literal scatter_indices = LiteralUtil::CreateR2<int32>( local
2854 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 2}}); local
[all...] |
| hlo_parser_test.cc | 868 ENTRY %Scatter (input_tensor: f32[50,49,48,47,46], scatter_indices: s64[10,9,8,7,5], updates: f32[10,9,8,7,30,29,28,27,26]) -> f32[50,49,48,47,46] {
870 %scatter_indices = s64[10,9,8,7,5]{4,3,2,1,0} parameter(1)
872 ROOT %scatter = f32[50,49,48,47,46]{4,3,2,1,0} scatter(f32[50,49,48,47,46]{4,3,2,1,0} %input_tensor, s64[10,9,8,7,5]{4,3,2,1,0} %scatter_indices, f32[10,9,8,7,30,29,28,27,26]{8,7,6,5,4,3,2,1,0} %updates), update_window_dims={4,5,6,7,8}, inserted_window_dims={}, scatter_dims_to_operand_dims={0,1,2,3,4}, index_vector_dim=4, to_apply=%add_F32.v3
[all...] |
| hlo_evaluator_typed_visitor.h | [all...] |
| hlo_instruction_test.cc | 1434 HloInstruction* scatter_indices = local
[all...] |
| hlo_instruction.h | [all...] |
| hlo_instructions.h | [all...] |
| hlo_instruction.cc | [all...] |
| hlo_instructions.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| local_computation_builder.h | 377 LocalOp Scatter(const LocalOp& input, const LocalOp& scatter_indices,
|
| local_computation_builder.cc | 763 const LocalOp& input, const LocalOp& scatter_indices,
766 return xla::Scatter(input.op(), scatter_indices.op(), updates.op(),
|
| xla_client_test.py | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| ir_emitter_unnested.cc | 1031 const HloInstruction* scatter_indices = scatter->operand(1); local
1079 const HloInstruction* scatter_indices = scatter->operand(1); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| xla_builder.h | 547 XlaOp Scatter(const XlaOp& input, const XlaOp& scatter_indices,
[all...] |
| xla_builder.cc | [all...] |
