Home
last modified time | relevance | path

Searched refs:shape_tuple (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python_api/
Dxla_shape.py147 def CreateShapeFromDtypeAndTuple(dtype, shape_tuple): # pylint: disable=invalid-name argument
158 return Shape(element_type, shape_tuple)
/external/tensorflow/tensorflow/python/autograph/utils/
Dtensor_list_test.py34 def _shape(self, shape_tuple): argument
35 return constant(shape_tuple, dtypes.int32)
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py585 shape_tuple = grad._shape_tuple() # pylint: disable=protected-access
586 if shape_tuple is None or None in shape_tuple:
588 return functools.reduce(operator.mul, shape_tuple, 1)
Dpywrap_tfe_src.cc1495 PyObject* shape_tuple = PyObject_CallMethod(tensor, _shape_tuple, nullptr); local
1501 if (ListContainsNone(shape_tuple)) {
1505 auto l = MakeIntList(shape_tuple);
1506 Py_DECREF(shape_tuple);
2935 tensorflow::Safe_PyObjectPtr shape_tuple( local
2938 if (shape_tuple == nullptr) {
2943 if (shape_tuple.get() == Py_None) {
2951 shape_tuple.get(), "shape_tuple didn't return a sequence"));