Searched refs:add_binary (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 165 auto add_binary = [&](const Shape& shape, const HloOpcode opcode, in HandleBatchNormTraining() local
212 add_binary(operand_shape, HloOpcode::kMultiply, operand, operand); in HandleBatchNormTraining()
234 add_binary(feature_shape, HloOpcode::kMultiply, mean, mean); in HandleBatchNormTraining()
238 add_binary(feature_shape, HloOpcode::kSubtract, square_mean, mean_square); in HandleBatchNormTraining()
245 add_binary(operand_shape, HloOpcode::kAdd, var_broadcasted, epsilon); in HandleBatchNormTraining()
251 auto operand_minus_mean = add_binary(operand_shape, HloOpcode::kSubtract, in HandleBatchNormTraining()
255 auto normalized = add_binary(operand_shape, HloOpcode::kMultiply, in HandleBatchNormTraining()
259 auto scaled_normalized = add_binary(operand_shape, HloOpcode::kMultiply, in HandleBatchNormTraining()
263 auto shifted_normalized = add_binary(operand_shape, HloOpcode::kAdd, in HandleBatchNormTraining()
333 auto add_binary = [&](const Shape& shape, const HloOpcode opcode, in HandleBatchNormInference() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | compiled_program_cache_generated.h | 58 void add_binary(flatbuffers::Offset<flatbuffers::Vector<uint8_t>> binary) { in add_binary() function
79 builder_.add_binary(binary);
|
| D | program_cache.cc | 132 program_builder.add_binary(binary_offset); in GetSerializedCache()
|