| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| cwise_ops.cc | 34 const TensorShape rhs_shape = ctx->InputShape(1); local
42 BCast bcast(BCast::FromShape(lhs_shape), BCast::FromShape(rhs_shape));
46 rhs_shape.DebugString()));
67 int max_rank = std::max(lhs_shape.dims(), rhs_shape.dims());
68 int min_rank = std::min(lhs_shape.dims(), rhs_shape.dims());
80 rhs_shape.dim_sizes(), bcast, extend_dimension);
|
| cwise_ops.h | 49 // (lhs_shape,rhs_shape) are their respective
62 const gtl::ArraySlice<int64>& rhs_shape, const BCast& broadcast_helper,
|
| strided_slice_op.cc | 260 const TensorShape rhs_shape = ctx->InputShape(4); variable
271 if (final_shape.num_elements() == 0 && rhs_shape.num_elements() == 0) {
273 // check that rhs_shape can be broadcast to final_shape, but that is
280 OP_REQUIRES(ctx, final_shape == rhs_shape,
283 " does not match r-value shape ", rhs_shape.DebugString(),
|
| binary_ops.cc | 41 const gtl::ArraySlice<int64>& rhs_shape, \
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gemm_thunk.h | 43 const Shape& lhs_shape, const Shape& rhs_shape,
|
| ir_emission_utils.cc | 47 bool AreValidGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape,
55 IsRank2WithNoPadding(rhs_shape) &&
58 !ShapeUtil::HasZeroElements(rhs_shape);
71 const Shape& rhs_shape = hlo.operand(1)->shape(); local
75 if (AreValidGemmShapes(lhs_shape, rhs_shape, hlo.shape())) {
79 CHECK_EQ(lhs_shape.dimensions(1), rhs_shape.dimensions(0));
|
| cudnn_convolution_algorithm_picker.cc | 269 const auto& rhs_shape = instr->operand(1)->shape(); local
275 /*filter_shape=*/rhs_shape, /*output_shape=*/conv_result_shape,
280 /*filter_shape=*/rhs_shape, /*output_shape=*/lhs_shape, instr->window(),
285 /*filter_shape=*/conv_result_shape, /*output_shape=*/rhs_shape,
|
| gemm_thunk.cc | 213 const Shape& lhs_shape, const Shape& rhs_shape,
221 rhs_shape_(rhs_shape),
|
| ir_emitter.cc | 449 const Shape& rhs_shape = rhs_instruction->shape(); local
451 if (ShapeUtil::IsScalar(lhs_shape) && ShapeUtil::IsScalar(rhs_shape)) {
484 !ShapeUtil::IsScalar(rhs_shape));
493 ShapeUtil::Rank(rhs_shape) >= 2
494 ? ShapeUtil::GetDimensionNumber(rhs_shape, -2)
499 rhs_shape.dimensions(rhs_reduction_dimension));
[all...] |
| ir_emitter_unnested.cc | 414 const auto& rhs_shape = custom_call->operand(1)->shape(); local
442 /*filter_shape=*/rhs_shape,
455 /*filter_shape=*/rhs_shape,
469 /*output_shape=*/rhs_shape, //
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| dot_decomposer.cc | 38 const Shape& rhs_shape = rhs->shape(); local
50 rhs_shape.dimensions(dnums.rhs_batch_dimensions(i)));
75 PrimitiveType rhs_type = rhs_shape.element_type();
76 const int64 rhs_rows = rhs_shape.dimensions(num_batch_dims + 0);
77 const int64 rhs_cols = rhs_shape.dimensions(num_batch_dims + 1);
|
| shape_inference_test.cc | 405 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 3}); local
427 ShapeInference::InferConvolveShape(lhs_shape, rhs_shape, window, dnums);
449 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 3}); local
472 ShapeInference::InferConvolveShape(lhs_shape, rhs_shape, window, dnums);
494 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 4}); local
517 ShapeInference::InferConvolveShape(lhs_shape, rhs_shape, window, dnums);
527 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 11, 3, 2}); local
555 ShapeInference::InferConvolveShape(lhs_shape, rhs_shape, window, dnums);
987 Shape rhs_shape = ShapeUtil::MakeShape(F32, {5, 2, 3, 14}); local
1012 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 3, 14}); local
1034 Shape rhs_shape = ShapeUtil::MakeShape(F32, {3, 3, 14}); local
1053 Shape rhs_shape = ShapeUtil::MakeShape(F32, {3, 2, 14}); local
1072 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 3, 14}); local
1091 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 3, 14}); local
[all...] |
| hlo_cost_analysis.cc | 203 const Shape& rhs_shape = dot->operand(1)->shape(); local
216 ShapeUtil::ElementsIn(rhs_shape);
|
| algebraic_simplifier_test.cc | 2585 Shape rhs_shape = ShapeUtil::MakeShape(F32, {k, n}); local
2671 Shape rhs_shape = ShapeUtil::MakeShape(F32, {spec.k, spec.n}); local
2734 Shape rhs_shape = ShapeUtil::MakeShape(F32, {spec.k, spec.n}); local
[all...] |
| hlo_evaluator.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_layout_assignment_test.cc | 60 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 24}); local
65 HloInstruction::CreateConstant(Literal::CreateFromShape(rhs_shape)));
95 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 24}); local
102 HloInstruction::CreateConstant(Literal::CreateFromShape(rhs_shape)));
138 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1}); local
146 HloInstruction::CreateConstant(Literal::CreateFromShape(rhs_shape)));
179 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1}); local
184 HloInstruction::CreateParameter(0, rhs_shape, "param0"));
193 ShapeLayout(LayoutUtil::GetWithDefaultLayout(rhs_shape));
212 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1}) local
[all...] |
| dot_op_emitter.cc | 719 const Shape& rhs_shape = rhs_array_.GetShape(); local
721 if (ShapeUtil::IsScalar(lhs_shape) || ShapeUtil::IsScalar(rhs_shape)) {
724 ShapeUtil::IsScalar(rhs_shape));
747 if (ShapeUtil::Rank(rhs_shape) >= 2) {
749 ShapeUtil::GetDimensionNumber(rhs_shape, transpose_rhs_ ? -1 : -2);
754 rhs_shape.dimensions(rhs_reduction_dimension));
759 rhs_reduction_dimension == LayoutUtil::Minor(rhs_shape.layout(), 0);
1003 const Shape& rhs_shape = rhs_array_.GetShape(); local
1063 const Shape& rhs_shape = hlo.operand(1)->shape(); local
1090 const Shape& rhs_shape = dot->operand(1)->shape(); local
[all...] |
| cpu_layout_assignment.cc | 146 Shape rhs_shape(RowMajorShape(rhs_instruction->shape()));
147 TF_RETURN_IF_ERROR(constraints->SetOperandLayout(rhs_shape, dot, 1));
|
| ir_emitter.cc | 859 const Shape& rhs_shape = rhs->shape(); local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_ops_test.cc | 135 TensorShape rhs_shape; local
136 rhs_shape = TensorShape({cols});
137 Tensor rhs(type, rhs_shape);
236 TensorShape rhs_shape; local
238 rhs_shape = TensorShape({rows, 1});
240 rhs_shape = TensorShape({cols});
242 Tensor rhs(DT_FLOAT, rhs_shape);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| matrix_ops_simple_test.cc | 178 Shape rhs_shape = local
196 auto rhs_arg = builder.Parameter(1, rhs_shape, "rhs");
244 Shape rhs_shape = local
264 auto rhs_arg = builder.Parameter(1, rhs_shape, "rhs");
|
| xla_hlo_profile_test.cc | 181 Shape rhs_shape = ShapeUtil::MakeShape(F32, {m, k}); local
197 rhs_shape);
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.cc | 268 std::unique_ptr<Shape> rhs_shape = GetShape(rhs).ConsumeValueOrDie(); local
270 << ShapeUtil::HumanString(*rhs_shape);
271 CHECK(ShapeUtil::Equal(*lhs_shape, *rhs_shape))
273 << ShapeUtil::HumanString(*rhs_shape);
575 const Shape& lhs_shape, const Shape& rhs_shape,
577 if (ShapeUtil::Rank(lhs_shape) != ShapeUtil::Rank(rhs_shape)) {
582 ShapeUtil::HumanString(rhs_shape).c_str()));
591 ShapeUtil::HumanString(rhs_shape).c_str()));
646 std::unique_ptr<Shape> rhs_shape = rhs_shape_or_status.ConsumeValueOrDie(); local
648 if (!VerifyConvolution(*lhs_shape, *rhs_shape, dimension_numbers))
705 std::unique_ptr<Shape> rhs_shape = rhs_shape_or_status.ConsumeValueOrDie(); local
[all...] |
| computation_builder.h | [all...] |
