Home
last modified time | relevance | path

Searched refs:OnesLike (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DOnesLike.pbtxt2 name: "OnesLike"
27 name: "OnesLike"
58 name: "OnesLike"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DOnesLike.pbtxt2 name: "OnesLike"
27 name: "OnesLike"
58 name: "OnesLike"
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_OnesLike.pbtxt2 graph_op_name: "OnesLike"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_OnesLike.pbtxt2 graph_op_name: "OnesLike"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_OnesLike.pbtxt2 graph_op_name: "OnesLike"
/external/tensorflow/tensorflow/c/experimental/ops/
Darray_ops.h45 Status OnesLike(AbstractContext* ctx,
Darray_ops.cc84 Status OnesLike(AbstractContext* ctx, in OnesLike() function
/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/
Dlarge_function_graph.pbtxt283 op: "OnesLike"
377 op: "OnesLike"
/external/tensorflow/tensorflow/c/experimental/gradients/
Dmath_grad.cc31 using tensorflow::ops::OnesLike;
351 TF_RETURN_IF_ERROR(OnesLike(ctx, {Conj_X.get()}, in Compute()
/external/tensorflow/tensorflow/cc/framework/
Dgradients_test.cc35 using ops::OnesLike;
127 auto dz = OnesLike(scope, z); in TEST_F()
Dgradients.cc575 grad_inputs.emplace_back(ops::OnesLike(scope, output)); in AddSymbolicGradients()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc1174 PyObject* OnesLike() const;
1292 PyObject* OnesLike(PyObject* tensor) const { in OnesLike() function in PyVSpace
1302 *result = t.OnesLike(); in BuildOnesLike()
1438 PyObject* PyTapeTensor::OnesLike() const { in OnesLike() function in PyTapeTensor
1441 return py_vspace->OnesLike(tensor); in OnesLike()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md295 * `raw_ops.OnesLike`
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md154 `OnesLike` | `T={bool,complex64,double,float,int32,int64}`
Dcpu_supported_ops.md154 `OnesLike` | `T={bool,complex64,double,float,int32,int64}`
/external/tensorflow/tensorflow/cc/gradients/
Dmath_grad.cc658 auto axes_ones = OnesLike(scope, axes); in ReducedShapeHelper()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1737 const3 = OnesLike(expected_graph_, s_, matmul, "gradients/OnesLike"); in BuildExpectedGraph()
1784 TF_Operation* OnesLike(TF_Graph* graph, TF_Status* s, TF_Operation* in, in OnesLike() function in tensorflow::__anon4051605d0111::CApiGradientsTest
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc786 Output ones_like = ops::OnesLike(s.WithOpName("ones_like"), x); in TEST_F()
3991 Output ones_like = ops::OnesLike(scope.WithOpName("ones_like"), x); in TEST_F()
4039 Output ones_like = ops::OnesLike(scope.WithOpName("ones_like"), x); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc2060 Output a1 = ops::OnesLike(s.WithOpName("a1"), a); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlower_tf.mlir552 %0 = "tf.OnesLike"(%arg0) : (tensor<*xi32>) -> tensor<*xi32>
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2479 TEST_F(OpTest, OnesLike) { in TEST_F() argument
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt2692 name: "OnesLike"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt2692 name: "OnesLike"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt41335 name: "OnesLike"
41360 name: "OnesLike"
41391 name: "OnesLike"
/external/tensorflow/
DRELEASE.md3021 * ZerosLike and OnesLike ops treated as constants by Graph Transform Tool.

12