| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | concat_test.cc | 42 ConcatInDim(&builder, {}, 0); in XLA_TEST_F()
53 ConcatInDim(&builder, {a}, 0); in XLA_TEST_F()
62 ConcatInDim(&builder, {a}, 0); in XLA_TEST_F()
74 ConcatInDim(&builder, {a, b}, 0); in XLA_TEST_F()
85 ConcatInDim(&builder, {a, b}, 0); in XLA_TEST_F()
95 ConcatInDim(&builder, {a, b}, 0); in XLA_TEST_F()
105 ConcatInDim(&builder, {a, b}, 0); in XLA_TEST_F()
115 ConcatInDim(&builder, {a, b}, 0); in XLA_TEST_F()
135 ConcatInDim(&builder, {a, b}, 0); in XLA_TEST_F()
145 ConcatInDim(&builder, {a, b}, dim); in XLA_TEST_F()
[all …]
|
| D | dot_operation_test.cc | 574 auto out_flat = ConcatInDim(&builder, out_slices, 0); in XLA_TYPED_TEST()
831 ConcatInDim(&builder, {rhs_arg_0, rhs_arg_1, rhs_arg_2}, 0)); in XLA_TYPED_TEST()
878 Dot(ConcatInDim(&builder, {lhs_arg_0, lhs_arg_1, lhs_arg_2}, 1), in XLA_TYPED_TEST()
|
| D | while_test.cc | 921 auto inc = ConcatInDim(&builder, in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pack_op.cc | 82 ctx->SetOutput(0, xla::ConcatInDim(ctx->builder(), reshaped_inputs, axis)); in Compile()
|
| D | image_ops.cc | 127 context->SetOutput(0, xla::ConcatInDim(b, hsv, channel_dim)); in Compile()
163 context->SetOutput(0, xla::ConcatInDim(b, rgb, channel_dim)); in Compile()
266 xla::ConcatInDim(b, rgb, channel_dim), type); in Compile()
325 xla::ConcatInDim(b, rgb, channel_dim), type); in Compile()
|
| D | cross_op.cc | 76 auto output = xla::ConcatInDim(b, {s1, s2, s3}, in0_shape.dims() - 1); in Compile()
|
| D | mirror_pad_op.cc | 59 accum = xla::ConcatInDim(b, {lhs_pad, accum, rhs_pad}, dimno); in DoMirrorPad()
|
| D | reverse_sequence_op.cc | 97 xla::ConcatInDim(builder, {batch_idx, reverse_idx}, in Compile()
|
| D | stateful_random_ops.cc | 64 auto result = ConcatInDim(builder, outputs, 0); in StatefulRngUniformU32()
171 return ConcatInDim( in ConcatScalars()
|
| D | concat_op.cc | 93 ctx->SetOutput(0, xla::ConcatInDim(ctx->builder(), input_data, axis)); in Compile()
|
| D | dynamic_stitch_op.cc | 190 ctx->SetOutput(0, xla::ConcatInDim(ctx->builder(), to_concat, 0)); in Compile()
|
| D | image_resize_ops.cc | 290 input_data = xla::ConcatInDim(builder, {input_data, slice}, 1); in ResizeUsingDilationAndConvolution()
300 input_data = xla::ConcatInDim(builder, {input_data, slice}, 2); in ResizeUsingDilationAndConvolution()
|
| D | resampler_ops.cc | 131 return xla::ConcatInDim(b, {batch_indices, indices}, dimensions.size() - 1); in ConcatenateIota()
479 return xla::ConcatInDim(ctx->builder(), {reshaped_x, reshaped_y}, in CalculateGradWarp()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | comparators_test.cc | 77 ConcatInDim(test->builder(), all_comparisons, 0); in BuildComparatorAndComparisons()
|
| D | prng.cc | 125 auto result = ConcatInDim(builder, outputs, 0); in StatelessRngUniformU32()
|
| D | slicing.cc | 152 XlaOp gather_indices = ConcatInDim(builder, to_concat, input_shape.rank()); in TorchGather()
|
| D | math_test.cc | 132 ConcatInDim( in TestSqrtPowInequivalence()
|
| D | svd.cc | 781 result.u = ConcatInDim(builder, in SortBySingularValuesAndPostProcessing()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 540 LocalOp ComputationBuilder::ConcatInDim(absl::Span<const LocalOp> operands, in ConcatInDim() function in xla::swig::ComputationBuilder
547 return xla::ConcatInDim(&builder_, xla_ops, dimension); in ConcatInDim()
|
| D | local_computation_builder.h | 275 LocalOp ConcatInDim(absl::Span<const LocalOp> operands, int64 dimension);
|
| D | local_computation_builder.i | 311 %unignore xla::swig::ComputationBuilder::ConcatInDim;
|
| D | xla_client.py | 1128 return self._client.ConcatInDim(operands, dimension)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 114 ConcatInDim(builder, {diag_blocks, last_blocks}, ndims - 2); in DiagonalBlocks()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 366 XlaOp ConcatInDim(absl::Span<const XlaOp> operands, int64 dimension);
752 friend XlaOp ConcatInDim(XlaBuilder* builder,
1247 XlaOp ConcatInDim(XlaBuilder* builder, absl::Span<const XlaOp> operands,
|
| D | xla_builder.cc | 852 XlaOp XlaBuilder::ConcatInDim(absl::Span<const XlaOp> operands, in ConcatInDim() function in xla::XlaBuilder
2210 return this->ConcatInDim(received, concat_dimension); in AllToAll()
2900 XlaOp ConcatInDim(XlaBuilder* builder, absl::Span<const XlaOp> operands, in ConcatInDim() function
2902 return builder->ConcatInDim(operands, dimension); in ConcatInDim()
|