Searched refs:diff_scale_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 1280 Tensor* diff_scale_tensor = nullptr; in Compute() local
1282 AllocateTFOutputs(context, scale_tensor.shape(), &diff_scale_tensor, in Compute()
1286 auto diff_scale_data = diff_scale_tensor->flat<U>().data(); in Compute()
1329 Tensor* diff_scale_tensor = nullptr; in HandleEmptyInput() local
1331 AllocateTFOutputs(context, tf_shape_scale_shift, &diff_scale_tensor, in HandleEmptyInput()
1337 Tensor** diff_scale_tensor, in AllocateTFOutputs() argument
1339 DCHECK(diff_scale_tensor); in AllocateTFOutputs()
1350 AllocateOutputSetMklShape(context, kDiffScaleIndex, diff_scale_tensor, in AllocateTFOutputs()
1353 DCHECK(*diff_scale_tensor); in AllocateTFOutputs()
1355 auto diff_scale_data = (*diff_scale_tensor)->flat<U>().data(); in AllocateTFOutputs()
[all …]
|