| /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 a_nnz, in UnionSparseIndicesAndValues()
153 const auto a_values = a_values_t->vec<T>(); in Compute() local
|
| D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 32 const Tindices* __restrict__ a_indices, const T* __restrict__ a_values, in SparseTensorDenseMatMulKernel()
|
| D | sparse_tensor_dense_add_op.cc | 35 Status ValidateInputs(const Tensor *a_indices, const Tensor *a_values, in ValidateInputs()
|
| D | sparse_add_op.cc | 57 auto a_values = ctx->input(1).vec<T>(); in Compute() local
|
| D | sparse_tensor_dense_matmul_op.cc | 44 const Tensor* a_values; in Compute() local
|
| /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 | 576 auto a_values = a_matrix->values_vec<T>(i); in Compute() local
666 auto a_values = a_input_matrix->values_vec<T>(i); in Compute() local
|
| /external/tensorflow/tensorflow/lite/micro/memory_planner/ |
| D | greedy_memory_planner_test.cc | 38 int a_values[a_size] = {10, 9, 8, 7, 6, 5, 4, 3, 2, 1}; in TF_LITE_MICRO_TEST() local
|
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
| D | device_tracer_test.cc | 77 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 | 94 void Initialize(std::initializer_list<float> a_values) { in Initialize()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | array_elementwise_ops_test.cc | 348 std::vector<float> a_values; in TEST_P() local
|