Home
last modified time | relevance | path

Searched refs:mat_mul (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/framework/experimental/
Dmath_ops.py30 def mat_mul(a, b, name=None): function
32 return _math_ops.mat_mul(ctx, a, b, name)
Dunified_api_test.py415 x = math_ops_lib.mat_mul(x, hidden_weight)
417 logits = math_ops_lib.mat_mul(x, softmax_weight)
/external/tensorflow/tensorflow/python/ops/
Dmath_grad.py1689 grad_a = gen_math_ops.mat_mul(grad, b, transpose_b=True)
1691 grad_a = gen_math_ops.mat_mul(grad, b)
1693 grad_a = gen_math_ops.mat_mul(b, grad, transpose_b=True)
1695 grad_a = gen_math_ops.mat_mul(b, grad, transpose_a=True, transpose_b=True)
1705 grad_b = gen_math_ops.mat_mul(a, grad, transpose_a=True)
1707 grad_b = gen_math_ops.mat_mul(grad, a, transpose_a=True)
1709 grad_b = gen_math_ops.mat_mul(a, grad)
1711 grad_b = gen_math_ops.mat_mul(grad, a, transpose_a=True, transpose_b=True)
1734 grad_a = gen_math_ops.mat_mul(grad, b, transpose_b=True)
1735 grad_b = gen_math_ops.mat_mul(a, grad, transpose_a=True)
[all …]
Dmath_ops.py3451 return gen_math_ops.mat_mul(
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dops_defs.py162 grad_a = gen_math_ops.mat_mul(grad, b)
163 grad_b = gen_math_ops.mat_mul(grad, a, transpose_a=True)
/external/tensorflow/tensorflow/python/framework/
Dgraph_building_benchmark.py95 return gen_math_ops.mat_mul(x, y)
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.h238 MatMulDimensions* mat_mul,
Dop_level_cost_estimator.cc1040 MatMulDimensions* mat_mul, in CountMatMulOperations() argument
1102 if (mat_mul != nullptr) { in CountMatMulOperations()
1103 mat_mul->m = m_dim; in CountMatMulOperations()
1104 mat_mul->n = n_dim; in CountMatMulOperations()
1105 mat_mul->k = k_dim; in CountMatMulOperations()
/external/llvm-project/llvm/test/Analysis/Delinearization/
Dconstant_functions_multi_dim.ll21 define void @mat_mul(float* %C, float* %A, float* %B, i64 %N) #0 !kernel_arg_addr_space !2 !kernel_…
/external/llvm-project/polly/test/ScopInfo/
Dconstant_functions_multi_dim.ll58 define void @mat_mul(float* %C, float* %A, float* %B, i64 %N) #0 !kernel_arg_addr_space !2 !kernel_…
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_test.py363 math_ops.mat_mul([[1., 1.] * 2], [[1., 1.] * 3])
Dbenchmarks_test.py424 gen_math_ops.mat_mul(m, m, transpose_b=transpose_b)
/external/tensorflow/tensorflow/python/distribute/
Dstrategy_test_lib.py151 gen_math_ops.mat_mul(x, kernel), []) - array_ops.identity(1.)
210 gen_math_ops.mat_mul(x, kernel), []) - array_ops.identity(1.)
Dcollective_all_reduce_strategy_test.py129 gen_math_ops.mat_mul(x, kernel), []) - constant_op.constant(1.)
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_math_ops.py223 return gen_math_ops.mat_mul(x1, x2)