| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.cc | 169 for (auto& parameter_shape : *program_shape->mutable_parameters()) { in SetToDefaultLayout()
170 LayoutUtil::SetToDefaultLayout(¶meter_shape); in SetToDefaultLayout()
269 for (auto& parameter_shape : *program_shape->mutable_parameters()) { in ClearLayout()
270 LayoutUtil::ClearLayout(¶meter_shape); in ClearLayout()
308 for (auto& parameter_shape : program_shape.parameters()) { in HasLayout() local
309 if (!LayoutUtil::HasLayout(parameter_shape)) { in HasLayout()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_optimization_parameters_utils.cc | 374 shape_inference::ShapeHandle parameter_shape; in operator ()() local
375 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, ¶meter_shape)); in operator ()()
380 TF_RETURN_IF_ERROR(c->Merge(parameter_shape, accumulator_j_shape, &merged)); in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | computation_layout.h | 72 const Shape& parameter_shape(int64 param_no) const { in parameter_shape() function
|
| D | hlo_module.cc | 344 const Shape& parameter_shape = in CreateFromProto() local
347 parameter_shape)) in CreateFromProto()
352 << ", actual: " << ShapeUtil::HumanStringWithLayout(parameter_shape); in CreateFromProto()
|
| D | layout_assignment.cc | 581 body_layout.parameter_shape(0))); in AddMandatoryConstraints()
583 condition_layout.parameter_shape(0))); in AddMandatoryConstraints()
640 branch_computation_layout.parameter_shape(0), instruction, k + 1, in AddMandatoryConstraints()
645 best_branch_computation_layout.parameter_shape(0), instruction, in AddMandatoryConstraints()
2065 computed_computation_layout.parameter_shape(i), shape_index); in PropagateComputationLayouts()
2148 ShapeUtil::Compatible(entry_computation_layout_->parameter_shape(i), in Run()
|
| D | hlo_verifier.cc | 646 const Shape& parameter_shape = to_apply.parameters(i); in SameElementTypesForOperandsAndToApplyParameters() local
648 if (!ShapeUtil::SameElementType(parameter_shape, operand_shape)) { in SameElementTypesForOperandsAndToApplyParameters()
1302 ShapeUtil::ValidateShapeWithOptionalLayout(layout.parameter_shape(i))); in VerifyEntryComputationLayout()
1303 if (!ShapeUtil::Compatible(parameter->shape(), layout.parameter_shape(i))) { in VerifyEntryComputationLayout()
1309 ShapeUtil::HumanString(layout.parameter_shape(i))); in VerifyEntryComputationLayout()
|
| D | shape_inference.cc | 1187 const Shape& parameter_shape = to_apply.parameters(i); in InferMapShape() local
1189 if (!ShapeUtil::IsScalar(parameter_shape)) { in InferMapShape()
1193 i, ShapeUtil::HumanString(parameter_shape)); in InferMapShape()
1196 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(parameter_shape, in InferMapShape()
1201 i, ShapeUtil::HumanString(parameter_shape), in InferMapShape()
|
| D | service.cc | 856 executable->module_config().entry_computation_layout().parameter_shape( in Execute()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.h | 74 bool InputShapeMatches(const xla::Shape& parameter_shape,
|
| D | xrt_util.cc | 224 bool InputShapeMatches(const xla::Shape& parameter_shape, in InputShapeMatches() argument
250 return xla::ShapeUtil::ForEachSubshapeWithStatus(parameter_shape, in InputShapeMatches()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 83 !computation_layout.parameter_shape(i).is_static()); in GetDynamicInputInfo()
97 [&](int64 i) { return computation_layout.parameter_shape(i); }, in GetInputTuples()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 5209 Shape parameter_shape = TypeToShape(parameter.getType()); in NumInputsInvolveInOnlyElementwiseOps() local
5210 return ShapeUtil::SameDimensions(op_shape, parameter_shape) && in NumInputsInvolveInOnlyElementwiseOps()
5222 Shape parameter_shape = TypeToShape(parameter.getType()); in NumInputsWithMoreElementsThan() local
5223 return ShapeUtil::ElementsIn(parameter_shape) > num_elements; in NumInputsWithMoreElementsThan()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 537 const Shape& parameter_shape = program_shape.parameters(alias.param_number); in PopulateInputOutputAlias() local
538 if (!ShapeUtil::IndexIsValid(parameter_shape, alias.param_index)) { in PopulateInputOutputAlias()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3704 old_entry_layout.parameter_shape(i), in Run()
|