Home
last modified time | relevance | path

Searched refs:parameter_replication (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
H A Dif_conditional.hlotxt25 %arg0.1 = f32[] parameter(0), parameter_replication={false}, metadata={op_name="XLA_Args"}
100 %arg0 = f32[] parameter(0), parameter_replication={false}
101 %arg1 = f32[] parameter(1), parameter_replication={false}
102 %arg2 = f32[] parameter(2), parameter_replication={false}
142 %arg0 = f32[] parameter(0), parameter_replication={false}
H A Dexport_replicas.mlir13 // CHECK-NOT: parameter_replication={true}
14 // CHECK: %[[ARG1:.*]] = f32[16,16] parameter(1), parameter_replication={true}
H A Dimport.hlotxt540 %arg0.1 = f32[3,9]{1,0} parameter(0), parameter_replication={false}, metadata={op_name="XLA_Args"}
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_parser.cc452 bool ParseParameterReplication(ParameterReplication* parameter_replication);
1157 optional<ParameterReplication> parameter_replication; in ParseInstructionRhs() local
1160 &parameter_replication}; in ParseInstructionRhs()
1209 if (parameter_replication) { in ParseInstructionRhs()
1212 parameter_replication->replicated_at_leaf_buffers(); in ParseInstructionRhs()
3182 ParameterReplication* parameter_replication) { in ParseParameterReplication() argument
3191 parameter_replication->add_replicated_at_leaf_buffers(true); in ParseParameterReplication()
3193 parameter_replication->add_replicated_at_leaf_buffers(false); in ParseParameterReplication()
4224 ParameterReplication parameter_replication; in ParseAttributeHelper() local
4225 if (!ParseParameterReplication(&parameter_replication)) { in ParseAttributeHelper()
[all …]
H A Dhlo.proto275 xla.ParameterReplication parameter_replication = 61; field
H A Dhlo_instruction.cc439 if (!proto.parameter_replication().replicated_at_leaf_buffers().empty()) { in CreateFromProto()
441 proto.parameter_replication().replicated_at_leaf_buffers()); in CreateFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/
H A Dconstant-folding.mlir18 …%[[ARG_TUPLE]] = (f32[10,19]{1,0}, f32[19,10]{1,0}) parameter(0), parameter_replication={false,tru…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dlaunch_dimensions.hlo97 …%arg0.1 = f16[6,512,4096]{2,1,0} parameter(0), parameter_replication={false}, metadata={op_name="X…