Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Ddefault_gradient.py56 def ones_like(t): function
Darray_ops.py3155 def ones_like(tensor, dtype=None, name=None, optimize=True): function
/external/tensorflow/tensorflow/c/eager/
Dtape.h619 Gradient* ones_like = nullptr; in InitialGradients() local
637 Gradient* ones_like = nullptr; in InitialGradients() local
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_array_ops.py277 def ones_like(tensor, dtype=None, name=None, optimize=True): function
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py78 def ones_like(a, dtype=None): function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc786 Output ones_like = ops::OnesLike(s.WithOpName("ones_like"), x); in TEST_F() local
3996 Output ones_like = ops::OnesLike(scope.WithOpName("ones_like"), x); in TEST_F() local
4044 Output ones_like = ops::OnesLike(scope.WithOpName("ones_like"), x); in TEST_F() local
/external/tensorflow/tensorflow/python/keras/
Dbackend.py1700 def ones_like(x, dtype=None, name=None): function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc4313 auto ones_like = ops::OnesLike(s.WithOpName("Ones"), input); in TEST_P() local