Searched refs:batch_norm_op (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_norm_benchmark.py | 40 def batch_norm_op(tensor, mean, variance, beta, gamma, scale): function
108 tensor = batch_norm_op(tensor, mean, variance, beta, gamma, scale)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 15 "batch_norm_op.cc",
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 273 #define MakeAndReturnBatchNormOp(batch_norm_op) \ in ImportInstructionImpl() argument
280 MakeAndReturn(batch_norm_op); \ in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 1525 ":batch_norm_op",
3880 ":batch_norm_op",
3922 name = "batch_norm_op",
3923 prefix = "batch_norm_op",
5898 "batch_norm_op.h",
5994 "batch_norm_op.cc",
6990 ":batch_norm_op",
|