| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 70 xla::BroadcastInDim(first_term, broadcast_dims, last_two_dims_indices); in BilinearWeights()
81 xla::BroadcastInDim(ratio, broadcast_dims, ratio_broadcast_indices); in BilinearWeights()
92 xla::BroadcastInDim(sign_change, broadcast_dims, last_two_dims_indices); in BilinearWeights()
225 xla::BroadcastInDim(is_in_bound, result_dims, broadcasted_dims); in BoundSamples()
289 auto broadcast_reshaped_weights = xla::BroadcastInDim( in CalculateGradData()
295 XlaOp broadcast_grad_output = xla::BroadcastInDim( in CalculateGradData()
411 xla::BroadcastInDim( in CalculateGradWarp()
419 xla::BroadcastInDim( in CalculateGradWarp()
427 xla::BroadcastInDim( in CalculateGradWarp()
435 xla::BroadcastInDim( in CalculateGradWarp()
[all …]
|
| D | xla_broadcast_helper_op.cc | 93 lhs = xla::BroadcastInDim(lhs, broadcast_shape, broadcast_dims); in Compile()
95 rhs = xla::BroadcastInDim(rhs, broadcast_shape, broadcast_dims); in Compile()
|
| D | permute_op.cc | 78 keys = xla::BroadcastInDim(keys, {4, 2}, {0}); in Compile()
|
| D | image_resize_ops.cc | 180 xla::BroadcastInDim(make_kernel_func(builder, type, kernel_size[1]), in MakeGeneralResizeKernel()
199 return xla::BroadcastInDim(make_kernel_func(builder, type, kernel_size[dim]), in MakeGeneralResizeKernelInDim()
222 return xla::BroadcastInDim(input, broadcast_shape.dimensions(), in BroadcastSpatialDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | broadcast_simple_test.cc | 164 BroadcastInDim(ConstantR1<float>(&b, {1, 2}), {2, 2}, {1});
177 BroadcastInDim(ConstantR1<float>(&b, {1, 2}), {2, 2}, {0});
190 BroadcastInDim(ConstantR2<float>(&b, {{1.0, 5.0}, {2.0, 6.0}}), {2, 2, 2},
208 BroadcastInDim(ConstantR2<float>(&b, {{1.0, 5.0}, {2.0, 6.0}}), {2, 2, 2},
226 BroadcastInDim(ConstantR1<float>(&b, {1, 2}), {3, 2}, {1});
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 84 xla::BroadcastInDim(input, broadcast_shape, broadcast_dims); in BroadcastTo()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 302 TEST_F(XlaBuilderTest, BroadcastInDim) { in TEST_F() argument
305 BroadcastInDim(x, {2, 4, 3}, in TEST_F()
315 BroadcastInDim(x, {2, 3, 4}, in TEST_F()
584 BroadcastInDim(gte, /*out_dim_size=*/{3, 5, 4}, in TEST_F()
|
| D | xla_builder.h | 332 XlaOp BroadcastInDim(const XlaOp& operand,
718 friend XlaOp BroadcastInDim(
1137 XlaOp BroadcastInDim(const XlaOp& operand,
|
| D | xla_builder.cc | 671 XlaOp XlaBuilder::BroadcastInDim( in BroadcastInDim() function in xla::XlaBuilder
2843 XlaOp BroadcastInDim(const XlaOp& operand, in BroadcastInDim() function
2846 return operand.builder()->BroadcastInDim(operand, out_dim_size, in BroadcastInDim()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd.cc | 156 BroadcastInDim(Lt(sigma, eps), x_shape.dimensions(), broadcast_dims), v, in HouseRow()
223 BroadcastInDim(Lt(sigma, eps), x_shape.dimensions(), broadcast_dims), v, in HouseCol()
752 d = BroadcastInDim(d, dimensions, broadcast_dims); in SortBySingularValuesAndPostProcessing()
|
| D | self_adjoint_eig_test.cc | 110 auto vw = Mul(result.v, BroadcastInDim(result.w, out_dims, broadcast_dims)); in ComputeMatmulVWVt()
|
| D | self_adjoint_eig.cc | 349 result.w = BroadcastInDim(result.w, dimensions, broadcast_dims); in SortByEigenvalues()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 475 LocalOp ComputationBuilder::BroadcastInDim( in BroadcastInDim() function in xla::swig::ComputationBuilder
478 return xla::BroadcastInDim(operand.op(), out_dim_sizes, broadcast_dimensions); in BroadcastInDim()
|
| D | local_computation_builder.h | 243 LocalOp BroadcastInDim(const LocalOp& operand,
|
| D | local_computation_builder.i | 301 %unignore xla::swig::ComputationBuilder::BroadcastInDim;
|
| D | xla_client.py | 1104 def BroadcastInDim(self, operand, shape, broadcast_dimensions): member in ComputationBuilder
1116 return self._client.BroadcastInDim(operand, shape, broadcast_dimensions)
|
| D | xla_client_test.py | 1112 c.BroadcastInDim(c.Constant(NumpyArrayS32([1, 2])), [2, 2], [0])
1114 c.BroadcastInDim(c.Constant(NumpyArrayS32([1, 2])), [2, 2], [1])
|