| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| concat_test.cc | 42 auto concatenated = builder.ConcatInDim({}, 0);
53 auto concatenated = builder.ConcatInDim({a}, 0);
62 auto concatenated = builder.ConcatInDim({a}, 0);
74 auto concatenated = builder.ConcatInDim({a, b}, 0);
85 auto concatenated = builder.ConcatInDim({a, b}, 0);
95 auto concatenated = builder.ConcatInDim({a, b}, 0);
105 auto concatenated = builder.ConcatInDim({a, b}, 0);
115 auto concatenated = builder.ConcatInDim({a, b}, 0);
135 auto concatenated = builder.ConcatInDim({a, b}, 0);
145 auto concatenated = builder.ConcatInDim({a, b}, dim)
[all...] |
| dot_operation_test.cc | 507 auto out_flat = builder.ConcatInDim(out_slices, 0);
631 lhs_constant, builder.ConcatInDim({rhs_arg_0, rhs_arg_1, rhs_arg_2}, 0));
679 builder.ConcatInDim({lhs_arg_0, lhs_arg_1, lhs_arg_2}, 1), rhs_constant);
|
| while_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| pack_op.cc | 83 ctx->SetOutput(0, ctx->builder()->ConcatInDim(reshaped_inputs, axis));
|
| cross_op.cc | 75 auto output = b->ConcatInDim({s1, s2, s3}, in0_shape.dims() - 1);
|
| mirror_pad_op.cc | 43 accum = b->ConcatInDim({lhs_pad, accum, rhs_pad}, dimno);
|
| image_ops.cc | 115 context->SetOutput(0, b->ConcatInDim(hsv, channel_dim));
151 context->SetOutput(0, b->ConcatInDim(rgb, channel_dim));
245 context->SetOutput(0, b->ConcatInDim(rgb, channel_dim));
299 context->SetOutput(0, b->ConcatInDim(rgb, channel_dim));
|
| concat_op.cc | 99 ctx->SetOutput(0, ctx->builder()->ConcatInDim(input_data, axis));
|
| dynamic_stitch_op.cc | 181 ctx->SetOutput(0, ctx->builder()->ConcatInDim(to_concat, 0));
|
| gather_op.cc | 146 auto out_index = bodyb->ConcatInDim(out_index_vals, 0);
|
| stateless_random_ops.cc | 148 builder->Reshape(builder->ConcatInDim(outputs, 0), shape.dim_sizes());
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| local_computation_builder.cc | 384 ComputationDataHandle LocalComputationBuilder::ConcatInDim(
387 return builder_.ConcatInDim(operands, dimension);
|
| local_computation_builder.h | 182 ComputationDataHandle ConcatInDim(
|
| local_computation_builder.i | 891 %unignore xla::swig::LocalComputationBuilder::ConcatInDim;
|
| xla_client.py | 623 self._client.ConcatInDim(_unwrap_data_handles(operands), dimension))
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| triangular_solve.cc | 488 return bodyb->ConcatInDim(padded_starts, 0);
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.h | 290 ComputationDataHandle ConcatInDim(
[all...] |
| computation_builder.cc | 341 ComputationDataHandle ComputationBuilder::ConcatInDim(
[all...] |
