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

  /external/tensorflow/tensorflow/compiler/xla/tests/
batch_normalization_test.cc 639 std::unique_ptr<GlobalData> variance_data = local
654 variance_data.get()},
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
mkl_fused_batch_norm_op.cc 62 // variance_data: output data buffer of variances
64 T* mean_data, T* variance_data) {
76 context_.variance_mem->set_data_handle(static_cast<void*>(variance_data));
301 // variance_data: input data buffer of variance
306 void Execute(const T* src_data, const T* mean_data, const T* variance_data,
314 static_cast<void*>(const_cast<T*>(variance_data)));
697 auto variance_data = reinterpret_cast<T*>(saved_variance_data_tf); variable
701 batch_variance_data[k] = variance_data[k] * adjust_factor;
704 std::memcpy(batch_variance_data, variance_data, depth_ * sizeof(T));
1006 T* variance_data = static_cast<T*>( variable
    [all...]

Completed in 123 milliseconds