Home
last modified time | relevance | path

Searched defs:result_values (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensorMath.cpp180 auto result_values = result.values(); in unary_op_out() local
224 …const auto result_values = op(sparse.values(), scalar.squeeze()).to(at::result_type(sparse, scalar… in intersection_binary_op_with_wrapped_scalar() local
296 auto result_values = op(values); in get_result_tensor_for_unary_op() local
389 auto result_values = self.values().mul(other); in mul_scalar_sparse_csr() local
H A DSparseUnaryOps.cpp121 auto result_values = result_impl->values(); in coalesced_unary_ufunc_out() local
H A DSparseTensorMath.cpp328 Tensor result_values = result._values(); in floor_divide_out_sparse_zerodim() local
2058 Tensor result_values = result._values(); in conj_physical_out_sparse() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
H A Dbenchmark_mlir_function.cc131 llvm::SmallVector<RCReference<AsyncValue>> result_values(num_results); in RunJitRtBenchmark() local
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Dtest_base.py138 def _compareOutputToExpected(self, result_values, expected_values, argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorConversions.cpp1643 scalar_t* result_values) { in _compressed_to_block_compressed_cpu_kernel()
1781 Tensor result_values = input_values.new_zeros(values_shape); in _compressed_to_block_compressed_cpu() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc1128 std::vector<float> result_values; in fold() local
/aosp_15_r20/external/deqp/external/openglcts/modules/gl/
H A Dgl4cShaderSubroutineTests.cpp8825 glw::GLuint result_values[4] = {0}; in verifySubroutineUniformValuesForShaderStage() local