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

  /external/tensorflow/tensorflow/core/kernels/
gemm_functors.h 101 typename tensorflow::TTypes<const T1>::Matrix a_matrix(a, m, k);
109 a_matrix.contract(b_matrix, dim_pair);
  /external/tensorflow/tensorflow/core/grappler/costs/
op_level_cost_estimator.cc 698 auto& a_matrix = op_info.inputs(0); local
717 MaybeGetMinimumShape(a_matrix.shape(), 2, found_unknown_shapes);
841 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); local
    [all...]
  /external/eigen/Eigen/src/Eigenvalues/
SelfAdjointEigenSolver.h 400 ::compute(const EigenBase<InputType>& a_matrix, int options)
404 const InputType &matrix(a_matrix.derived());
    [all...]
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_blas.cc 2258 const DeviceMemory<T> &a_matrix = *a_ptrs_to_wrappers[b]; local
2395 const auto *a_matrix = local
    [all...]

Completed in 321 milliseconds