| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd.cc | 132 auto zeros = ZerosLike(x); in HouseRow()
160 beta = Select(Lt(Add(sigma, ZerosLike(beta), broadcast_dims), eps), in HouseRow()
161 ZerosLike(beta), beta); in HouseRow()
198 auto zeros = ZerosLike(x); in HouseCol()
227 beta = Select(Lt(Add(sigma, ZerosLike(beta), broadcast_dims), eps), in HouseCol()
228 ZerosLike(beta), beta); in HouseCol()
381 auto c = Select(Ge(Abs(pqs), eps), c_temp, ZerosLike(c_temp) + one); in MakeJacobi()
382 auto s = Select(Ge(Abs(pqs), eps), s_temp, ZerosLike(s_temp)); in MakeJacobi()
439 XlaOp zeros = ZerosLike(tmp); in GetOneSidedJacobiRotation()
743 auto zeros = ZerosLike(d); in SortBySingularValuesAndPostProcessing()
|
| D | constants.h | 121 XlaOp ZerosLike(XlaOp prototype);
|
| D | matrix.cc | 110 return lower ? Select(TriangleMask(x, 0), x, ZerosLike(x)) in Triangle()
111 : Select(TriangleMask(x, -1), ZerosLike(x), x); in Triangle()
|
| D | constants.cc | 32 XlaOp ZerosLike(XlaOp prototype) { in ZerosLike() function
|
| D | constants_test.cc | 85 ZerosLike(ConstantR1<float>(&builder, {1., 2., 3.})); in XLA_TEST_F()
|
| D | self_adjoint_eig.cc | 99 auto c = Select(Ge(Abs(pqs), tol), c_temp, ZerosLike(c_temp) + one); in SymmetricShurDecomposition2x2()
100 auto s = Select(Ge(Abs(pqs), tol), s_temp, ZerosLike(s_temp)); in SymmetricShurDecomposition2x2()
|
| D | math.cc | 595 auto from_is_zero = Eq(from_abs, ZerosLike(from_abs)); in NextAfter()
596 auto to_is_zero = Eq(to_abs, ZerosLike(to_abs)); in NextAfter()
|
| D | svd_test.cc | 200 Add(result.d, ZerosLike(result.d)); in XLA_TEST_F()
|
| D | self_adjoint_eig_test.cc | 217 Add(result.w, ZerosLike(result.w)); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | binary_ops.cc | 75 auto zeros = xla::ZerosLike(x); in DivNoNanImpl()
93 auto zeros = xla::ZerosLike(x); in MulNoNanImpl()
134 auto zero = xla::ZerosLike(x); in XlogyImpl()
143 auto zero = xla::ZerosLike(x); in XdivyImpl()
|
| D | reverse_sequence_op.cc | 85 reverse_idx = xla::Select(xla::Lt(reverse_idx, xla::ZerosLike(reverse_idx)), in Compile()
|
| D | unary_ops.cc | 94 xla::Select(xla::Ne(x, x), xla::ZerosLike(x), xla::Sign(x)));
|
| D | conv_op_helpers.cc | 151 filter_backprop, xla::ZerosLike(filter_backprop)); in ContractFilterForDepthwiseBackprop()
|
| D | resampler_ops.cc | 555 auto is_ge_zero = xla::Ge(warp, xla::ZerosLike(warp)); in Compile()
|
| D | image_ops.cc | 502 auto square_zero = xla::ZerosLike(i_xmin); in Compile()
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_ZerosLike.pbtxt | 2 graph_op_name: "ZerosLike"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_ZerosLike.pbtxt | 2 graph_op_name: "ZerosLike"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ZerosLike.pbtxt | 2 graph_op_name: "ZerosLike"
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 65 XlaOp l = ZerosLike(a); in CholeskyUnblocked()
172 XlaOp l = ZerosLike(a); in BuildCholesky()
|
| D | triangular_solve_expander.cc | 270 auto x = ZerosLike(b); in SolveWithInvertedDiagonalBlocks()
419 a = lower ? Select(TriangleMask(a, -1), a, ZerosLike(a)) in BuildTriangularSolve()
420 : Select(TriangleMask(a, 0), ZerosLike(a), a); in BuildTriangularSolve()
|
| /external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/ |
| D | large_function_graph.pbtxt | 258 op: "ZerosLike"
352 op: "ZerosLike"
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 512 ZerosLike(scope, x)); in PowGrad()
520 ZerosLike(scope, x)); in PowGrad()
539 auto zeros = ZerosLike(scope, grad); in MaximumMinimumGradCommon()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients.cc | 510 dy[dy_index] = ops::ZerosLike(scope_, Output(n, dy_index)); in AddGradients()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 501 Output zeros_like = ops::ZerosLike(s.WithOpName("zeros_like"), x); in TEST_F()
753 Output zeros_known = ops::ZerosLike(s.WithOpName("zeros_known"), x_known); in TEST_F()
755 ops::ZerosLike(s.WithOpName("zeros_partially_known"), x_partially_known); in TEST_F()
757 ops::ZerosLike(s.WithOpName("zeros_unknown"), x_unknown); in TEST_F()
829 ops::ZerosLike(s.WithOpName("zeros_partially_known"), x_partially_known); in TEST_F()
831 ops::ZerosLike(s.WithOpName("zeros_unknown"), x_unknown); in TEST_F()
3645 Output zeros_like = ops::ZerosLike(scope.WithOpName("zeros_like"), x); in TEST_F()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 720 out_grads[i] = loop_state.ZerosLike(op, i)
|