HomeSort by relevance Sort by last modified time
    Searched refs:mirrored_weights (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
convolution_variants_test.cc 1155 auto mirrored_weights = Rev(weights, {2, 3}); local
1174 auto mirrored_weights = Rev(weights, {2, 3}); local
1194 auto mirrored_weights = Rev(weights, {2, 3}); local
1215 auto mirrored_weights = Rev(weights, {2, 3}); local
1307 auto mirrored_weights = Rev(weights, {2}); local
1350 auto mirrored_weights = Rev(weights, {2, 3, 4}); local
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
conv_op_helpers.cc 381 xla::XlaOp mirrored_weights = xla::Rev(filter, kernel_spatial_dims); local
384 // = gradients (with padding and dilation) <conv> mirrored_weights
386 out_backprop, mirrored_weights, /*window_strides=*/ones, padding,

Completed in 151 milliseconds