Home
last modified time | relevance | path

Searched refs:shape_tuple (Results 1 – 5 of 5) 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.py650 shape_tuple = grad._shape_tuple() # pylint: disable=protected-access
652 shape_tuple = grad.values._shape_tuple() # pylint: disable=protected-access
655 if shape_tuple is None or None in shape_tuple:
657 return functools.reduce(operator.mul, shape_tuple, 1)
Dpywrap_tfe_src.cc2135 tensorflow::Safe_PyObjectPtr shape_tuple( local
2142 if (ListContainsNone(shape_tuple.get()) || dtype == tensorflow::DT_VARIANT) {
2146 auto l = MakeIntList(shape_tuple.get());
4039 tensorflow::Safe_PyObjectPtr shape_tuple( local
4042 if (shape_tuple == nullptr) {
4047 if (shape_tuple.get() == Py_None) {
4055 shape_tuple.get(), "shape_tuple didn't return a sequence"));
Dbenchmarks_test.py1477 shape_tuple = x._shape_tuple()
1480 tensor_shape.TensorShape(shape_tuple)