| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensor_dense_matmul_op_test.cc | 26 Node* SparseTensorDenseMatMulNode(Graph* g, Node* a_indices, Node* a_values, in SparseTensorDenseMatMulNode()
45 Tensor a_values(DT_FLOAT, TensorShape({nnz})); in SparseTensorDenseMatmul() local
|
| D | sparse_sparse_binary_op_shared.cc | 64 typename TTypes<T>::ConstFlat a_values, int64_t a_nnz, in UnionSparseIndicesAndValues()
154 const auto a_values = a_values_t->vec<T>(); in Compute() local
|
| D | sparse_tensor_dense_matmul_op.cc | 44 const Tensor* a_values; in Compute() local
257 typename TTypes<T>::ConstVec a_values, typename TTypes<T>::ConstMatrix b) { in SparseTensorDenseMatMulImpl()
|
| D | sparse_add_op.cc | 63 auto a_values = ctx->input(1).vec<T>(); in Compute() local
|
| D | sparse_tensor_dense_add_op.cc | 36 Status ValidateInputs(const Tensor *a_indices, const Tensor *a_values, in ValidateInputs()
|
| D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 48 const Tindices* __restrict__ a_indices, const T* __restrict__ a_values, in SparseTensorDenseMatMulKernel()
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_2/ |
| D | multiplex_2_kernel.h | 77 const auto a_values = a_values_tensor.flat<T>(); in Compute() local
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_1/ |
| D | multiplex_1_kernel.cc | 61 const auto a_values = a_values_tensor.flat<T>(); in Compute() local
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_4/ |
| D | multiplex_4_kernel.cc | 101 const auto a_values = a_values_tensor.template flat<T>(); in Compute() local
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | GEMMMatrixMultiplyReshapedOnlyRhsMMUL.cpp | 62 const auto a_values = framework::dataset::make("alpha", {1.0f, 0.75f} ); variable
|
| D | GEMMMatrixMultiplyNative.cpp | 73 const auto a_values = framework::dataset::make("alpha", {1.0f, -0.75f} ); variable
|
| D | GEMMMatrixMultiplyReshapedOnlyRHS.cpp | 81 const auto a_values = framework::dataset::make("alpha", {-0.75f} ); variable
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mul_op.cc | 138 auto a_values = a.values().flat<T>(); in Compute() local
|
| D | kernels_gpu.cu.cc | 260 const T* a_values, const T* b_batch_values, T* c_values, in CSRSparseMatrixBatchMulVecKernel3D()
297 auto a_values = a.values().flat<T>(); in CSRSparseMatrixBatchMulVecImpl() local
|
| D | mat_mul_op.cc | 585 auto a_values = a_matrix->values_vec<T>(i); in Compute() local
675 auto a_values = a_input_matrix->values_vec<T>(i); in Compute() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_add_op_test.py | 210 a_values, argument
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/ |
| D | multiplex_3_kernel.cc | 78 const auto a_values = a_values_tensor.flat<T>(); in Compute() local
|
| /external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| D | device_tracer_test.cc | 84 void Initialize(std::initializer_list<float> a_values) { in Initialize()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | run_handler_test.cc | 598 void Initialize(std::initializer_list<float> a_values) { in Initialize()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 208 const std::vector<float>& a_values, in TestQuantizeMatMul()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_test.cc | 95 void Initialize(std::initializer_list<float> a_values) { in Initialize()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | array_elementwise_ops_test.cc | 352 std::vector<float> a_values; in TEST_P() local
|