Searched refs:packed_input (Results 1 – 3 of 3) sorted by relevance
92 packed_input = self.pack_uint8_r2_to_uint32(quantized_input.output)94 transposed_quantized_output = xla.dequantize(packed_input, 0.0, 255.0,
308 const auto& packed_input = packed_input_and_idx.value(); in SetReplicateOpOperands() local312 packed_input, region_arg_types[idx + num_replicated_block_args], in SetReplicateOpOperands()539 for (auto packed_input : packed_inputs) { in BuildReplicateOp() local540 state->addOperands(packed_input); in BuildReplicateOp()541 block.addArgument(packed_input.getType()); in BuildReplicateOp()
404 packed_input = ops.pack_eager_tensors([func_input])407 def func(packed_input): argument411 ret = packed_input + constant_op.constant(1.0)416 self.assertAllEqual(func(packed_input).numpy(),