Home
last modified time | relevance | path

Searched refs:t_a (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dsparse_csr_matrix_grad.py95 t_a = op.get_attr("transpose_a")
114 if not t_a:
118 grad_b = sparse_matmul(a, grad, transpose_a=not t_a, transpose_output=t_b)
119 elif not t_a and not t_b:
133 elif t_a and adj_b:
142 if not t_a:
147 a, grad, transpose_a=not t_a, transpose_b=True, transpose_output=t_b)
148 elif not t_a and not t_b:
163 elif t_a and adj_b:
175 t_a = op.get_attr("transpose_a")
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
Dcsr_sparse_matrix_test.py186 for (t_a, t_b, adj_a, adj_b) in itertools.product(*(([False, True],) * 4)):
187 if (t_a and adj_a) or (t_b and adj_b):
189 self._testSparseSparse(t_a, t_b, adj_a, adj_b)
224 for (t_a, t_b, adj_a, adj_b) in itertools.product(*(([False, True],) * 4)):
225 if (t_a and adj_a) or (t_b and adj_b):
227 self._testSparseDense(t_a, t_b, adj_a, adj_b)
262 for (t_a, t_b, adj_a, adj_b) in itertools.product(*(([False, True],) * 4)):
263 if (t_a and adj_a) or (t_b and adj_b):
265 self._testDenseSparse(t_a, t_b, adj_a, adj_b)
Dcsr_sparse_matrix_dense_mat_mul_grad_test.py111 for (t_a, t_b, adj_a, adj_b, t_out,
132 (dtype.__name__, t_a, t_b, adj_a, adj_b, t_out, conj_out))
136 create_mat_mul_test_fn(dtype, t_a, t_b, adj_a, adj_b, t_out, conj_out))
Dcsr_sparse_matrix_sparse_mat_mul_grad_test.py113 for (t_a, t_b, adj_a, adj_b) in itertools.product(*(([False, True],) * 4)):
131 "adj_a_%s_adj_b_%s" % (dtype.__name__, t_a, t_b, adj_a, adj_b))
134 create_sparse_mat_mul_test_fn(dtype, t_a, t_b, adj_a, adj_b))
/external/mesa3d/src/gallium/auxiliary/vl/
Dvl_deint_filter.c136 struct ureg_dst t_a, t_b; in create_deint_frag_shader() local
148 t_a = ureg_DECL_temporary(shader); in create_deint_frag_shader()
174 ureg_TEX(shader, t_a, TGSI_TEXTURE_2D_ARRAY, ureg_src(t_comp_bot), sampler_cur); in create_deint_frag_shader()
176 …ureg_ADD(shader, ureg_writemask(t_diff, TGSI_WRITEMASK_X), ureg_src(t_a), ureg_negate(ureg_src(t_b… in create_deint_frag_shader()
178 ureg_TEX(shader, t_a, TGSI_TEXTURE_2D_ARRAY, ureg_src(t_comp_top), sampler_prev); in create_deint_frag_shader()
180 …ureg_ADD(shader, ureg_writemask(t_diff, TGSI_WRITEMASK_Y), ureg_src(t_a), ureg_negate(ureg_src(t_b… in create_deint_frag_shader()
184 ureg_TEX(shader, t_a, TGSI_TEXTURE_2D_ARRAY, ureg_src(t_comp_top), sampler_cur); in create_deint_frag_shader()
186 …ureg_ADD(shader, ureg_writemask(t_diff, TGSI_WRITEMASK_X), ureg_src(t_a), ureg_negate(ureg_src(t_b… in create_deint_frag_shader()
188 ureg_TEX(shader, t_a, TGSI_TEXTURE_2D_ARRAY, ureg_src(t_comp_bot), sampler_prev); in create_deint_frag_shader()
190 …ureg_ADD(shader, ureg_writemask(t_diff, TGSI_WRITEMASK_Y), ureg_src(t_a), ureg_negate(ureg_src(t_b… in create_deint_frag_shader()
[all …]
/external/tensorflow/tensorflow/c/experimental/gradients/
Dmath_grad.cc125 bool t_a; in Compute() local
126 TF_RETURN_IF_ERROR(forward_attrs_.Get("transpose_a", &t_a)); in Compute()
150 if (!t_a && !t_b) { in Compute()
162 } else if (!t_a && t_b) { in Compute()
175 } else if (t_a && !t_b) { in Compute()
/external/webp/src/dsp/
Dupsampling_sse41.c54 const __m128i t_a = _mm_avg_epu8(a, da); /* (9a + 3b + 3c + d + 8) / 16 */ \
56 const __m128i t_1 = _mm_unpacklo_epi8(t_a, t_b); \
57 const __m128i t_2 = _mm_unpackhi_epi8(t_a, t_b); \
Dupsampling_sse2.c52 const __m128i t_a = _mm_avg_epu8(a, da); /* (9a + 3b + 3c + d + 8) / 16 */ \
54 const __m128i t_1 = _mm_unpacklo_epi8(t_a, t_b); \
55 const __m128i t_2 = _mm_unpackhi_epi8(t_a, t_b); \
/external/tensorflow/tensorflow/python/ops/
Dmath_grad.py1685 t_a = op.get_attr("transpose_a")
1688 if not t_a and not t_b:
1690 elif not t_a and t_b:
1692 elif t_a and not t_b:
1694 elif t_a and t_b:
1701 t_a = op.get_attr("transpose_a")
1704 if not t_a and not t_b:
1706 elif not t_a and t_b:
1708 elif t_a and not t_b:
1710 elif t_a and t_b:
[all …]
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.h474 const T& t_a = *a.get<T>(); in UnaryVariantBinaryOpRegistration()
477 return binary_op_fn(ctx, t_a, t_b, t_out); in UnaryVariantBinaryOpRegistration()
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_math_ops.py392 t_a = np_utils.cond(
411 a_shape = array_ops.shape(t_a)
413 a_reshaped = np_array_ops.reshape(t_a, _make_shape(a_shape, False))
/external/lz4/lib/
Dlz4hc.c889 struct { char c; LZ4_streamHC_t t; } t_a; in LZ4_streamHC_t_alignment() local
890 return sizeof(t_a) - sizeof(t_a.t); in LZ4_streamHC_t_alignment()
Dlz4.c1348 struct { char c; LZ4_stream_t t; } t_a; in LZ4_stream_t_alignment() local
1349 return sizeof(t_a) - sizeof(t_a.t); in LZ4_stream_t_alignment()