| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| reduction_ops_common.cc | 113 auto rx = xla::Parameter(&r, 0, xla::ShapeUtil::MakeShape(type, {}), "x");
114 auto ry = xla::Parameter(&r, 1, xla::ShapeUtil::MakeShape(type, {}), "y");
|
| reverse_sequence_op.cc | 78 xla::ShapeUtil::MakeShape(seq_lens_type, {batch_size, max_seq_len, 1}),
82 xla::ShapeUtil::MakeShape(seq_lens_type, {batch_size, max_seq_len, 1}),
|
| batchtospace_op.cc | 43 block_rank == xla::ShapeUtil::GetDimension(crops.shape(), 0) &&
44 2 == xla::ShapeUtil::GetDimension(crops.shape(), 1),
47 xla::ShapeUtil::HumanString(crops.shape())));
|
| spacetobatch_op.cc | 43 block_rank == xla::ShapeUtil::GetDimension(paddings.shape(), 0) &&
44 2 == xla::ShapeUtil::GetDimension(paddings.shape(), 1),
47 xla::ShapeUtil::HumanString(paddings.shape())));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| convolution_group_converter.cc | 170 Shape mask_shape = ShapeUtil::MakeShape(
192 Shape predicate_shape = ShapeUtil::MakeShape(
274 Shape shape = ShapeUtil::MakeShape(reduce_type, {});
359 ShapeUtil::DeleteDimension(kernel_input_feature_dim, filter->shape());
407 ShapeUtil::AppendMajorDimension(group_size, &reshaped_activation_shape);
418 ShapeUtil::AppendMajorDimension(1, &reshaped_filter_shape);
424 ShapeUtil::AppendMajorDimension(1, &new_output_shape);
449 ShapeUtil::DeleteDimension(new_spatial_dim, new_convolution->shape());
|
| dynamic_padder.cc | 154 ShapeUtil::ChangeElementType(operand->shape(), xla::U32);
156 ShapeUtil::ChangeElementType(operand->shape(), xla::PRED);
|
| hlo_input_output_alias_config_test.cc | 170 return ShapeUtil::ByteSizeOf(shape);
196 return ShapeUtil::ByteSizeOf(shape);
|
| hlo_ordering_test.cc | 57 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {});
130 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {});
143 ShapeUtil::MakeShape(xla::PRED, {}), cond_param));
180 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {});
213 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {});
226 ShapeUtil::MakeShape(xla::PRED, {}), cond_param));
411 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {});
459 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {});
|
| sort_simplifier.cc | 80 : ShapeUtil::MakeTupleShape(new_shapes);
91 ShapeUtil::MakeShape(sort->operand(i)->shape().element_type(), {});
|
| tuple_simplifier.cc | 90 if (!ShapeUtil::Compatible(top_tuple->shape(),
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| conv_canonicalization_test.cc | 85 ShapeUtil::MakeShape(
148 ShapeUtil::MakeShape(
|
| cpu_copy_insertion_test.cc | 62 const Shape scalar_shape_ = ShapeUtil::MakeShape(F32, {});
111 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_, scalar_shape_});
|
| cpu_instruction_fusion.cc | 26 return ShapeUtil::ByteSizeOfPrimitiveType(shape.element_type()) *
48 return !ShapeUtil::ElementIsComplex(hlo_shape) &&
|
| parallel_loop_emitter.cc | 38 CHECK(!ShapeUtil::IsScalar(shape_));
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| buffer_comparator.cc | 114 Shape shape = ShapeUtil::MakeShape(xla::F16, {num_elements});
136 Shape shape = ShapeUtil::MakeShape(xla::F16, {num_elements});
|
| gpu_fusible.cc | 131 (!ShapeUtil::Equal(instr_1->shape(), instr_2->shape()) ||
137 return ShapeUtil::EqualIgnoringFpPrecision(get_loop_shape(instr_1),
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| loop_emitter.cc | 81 CHECK(ShapeUtil::SameDimensions(shape_, array.GetShape()))
90 if (ShapeUtil::IsScalar(shape_)) {
|
| tuple_ops.cc | 40 CHECK(ShapeUtil::IsScalar(pred.GetShape()));
53 for (int i = 0; i < ShapeUtil::TupleElementCount(select.GetShape()); ++i) {
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| bitcast_convert_test.cc | 105 auto param = Parameter(b.get(), 0, ShapeUtil::MakeShape(F32, {}), "in");
117 auto param = Parameter(b.get(), 0, ShapeUtil::MakeShape(S32, {}), "in");
|
| check_execution_arity_test.cc | 80 auto p0 = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "param0");
81 auto p1 = Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {4}), "param1");
|
| compilation_cache_test.cc | 100 Neg(Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "param"));
155 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {2, 2}), "param0");
|
| multiple_devices_on_host_test.cc | 28 Shape scalar_s32 = ShapeUtil::MakeShape(S32, {});
95 ShapeUtil::MakeShape(S32, {}), device_ordinal));
|
| reverse_test.cc | 83 ShapeUtil::ElementsIn(ShapeUtil::MakeShape(F32, spec.input_dims)));
|
| compute_constant_test.cc | 131 ShapeUtil::MakeShape(F32, {}));
144 auto computation = Parameter(&b, 0, ShapeUtil::MakeShape(F32, {}), "param");
193 auto param_a = Parameter(&b, 10, ShapeUtil::MakeShape(F32, {}), "param0");
198 auto param_b = Parameter(&b, 1, ShapeUtil::MakeShape(F32, {}), "param1");
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| arithmetic.cc | 45 const Shape scalar = ShapeUtil::MakeShape(type, {});
156 ShapeUtil::ByteSizeOfPrimitiveType(output_type) * 8 - 1;
164 const int64 axis_size = ShapeUtil::GetDimension(input_shape, axis);
|
