Home
last modified time | relevance | path

Searched refs:_shape (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dtensor_spec.py51 self._shape = tensor_shape.TensorShape(shape)
66 return self._shape
81 self._shape.is_compatible_with(spec_or_value.shape))
103 shape = self._shape.most_specific_compatible_shape(other.shape)
108 return (self._shape, self._dtype, self._name)
114 return self._shape
160 "and shape %s." % (value, self._dtype, self._shape))
161 if not value.shape.is_compatible_with(self._shape):
163 "and shape %s." % (value, self._dtype, self._shape))
177 tensor_list[0].set_shape(self._shape)
[all …]
Dsparse_tensor.py278 self._shape = tensor_shape.as_shape(shape)
282 return (self._shape, self._dtype)
292 return self._shape
296 rank = self._shape.ndims
334 if self._shape.merge_with(dense_shape).ndims == 0:
344 rank = self._shape.ndims
349 if self._shape.is_fully_defined():
351 self._shape, dtype=dtypes.int64, name="shape")
352 elif (self._shape.rank is not None and
353 any(dim.value is not None for dim in self._shape.dims)):
[all …]
Dindexed_slices.py220 self._shape = tensor_shape.as_shape(shape)
230 return (self._shape, self._values_dtype, self._indices_dtype,
235 value_shape = self._indices_shape.concatenate(self._shape[1:])
241 tensor_spec.TensorSpec([self._shape.ndims], self._dense_shape_dtype))
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinear_operator_algebra_test.py52 def _shape(self): member in AdjointTest.testRegistration.CustomLinOp
94 def _shape(self): member in CholeskyTest.testRegistration.CustomLinOp
144 def _shape(self): member in MatmulTest.testRegistration.CustomLinOp
193 def _shape(self): member in SolveTest.testRegistration.CustomLinOp
239 def _shape(self): member in InverseTest.testRegistration.CustomLinOp
Dlinear_operator_test.py55 def _shape(self): member in LinearOperatorShape
82 def _shape(self): member in LinearOperatorMatmulSolve
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor.py503 spec_shape = typespec._shape # pylint: disable=protected-access
529 shape = typespec._shape
555 inner_shape=typespec._shape[typespec._ragged_rank + 1:])
605 self._shape = tensor_shape.as_shape(shape)
609 if self._shape.ndims is None:
628 return StructuredTensor(self._shape, components)
641 return (self._shape, self._field_specs)
646 tensor_shape.TensorShape([batch_size]).concatenate(self._shape),
652 self._shape[1:],
Dstructured_tensor_spec_test.py76 self.assertEqual(spec1._shape, (1, 2, 3))
81 self.assertEqual(spec2._shape, (1, 2))
/external/tensorflow/tensorflow/python/ops/
Dgradients_impl.py420 _shape = array_ops.shape(x)
422 array_ops.concat((_shape, _shape), 0))
Dresource_variable_ops.py419 self._shape = tensor_shape.as_shape(shape)
508 return self._shape
511 self._shape = self._shape.merge_with(shape)
813 handle=self._handle, dtype=self.dtype, shape=self._shape,
838 self._shape.assert_is_compatible_with(value_tensor.shape)
1674 self._shape = tensor_shape.TensorShape(
/external/tensorflow/tensorflow/python/autograph/utils/
Dtensor_list_test.py34 def _shape(self, shape_tuple): member in TensorListTest
43 l = list_ops.empty_tensor_list(self._shape(()), dtypes.int32)
53 l = tl.TensorList(self._shape(()), dtypes.int32)
/external/harfbuzz_ng/src/
Dhb-shape-plan.cc94 this->shaper_func = _hb_##shaper##_shape; \ in init()
118 else if (shapers[i].func == _hb_##shaper##_shape) \ in init()
400 _hb_##shaper##_shape (shape_plan, font, buffer, features, num_features); \ in hb_shape_plan_execute()
406 else if (shape_plan->key.shaper_func == _hb_##shaper##_shape) \ in hb_shape_plan_execute()
Dhb-shaper.cc33 #define HB_SHAPER_IMPLEMENT(name) {#name, _hb_##name##_shape},
Dhb-shaper.hh40 extern "C" HB_INTERNAL hb_shape_func_t _hb_##name##_shape;
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py90 self._shape = tensor_shape.TensorShape([first_dim] + first_var.shape[1:])
117 return self._shape
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py2153 self._shape = tensor_shape.as_shape(shape)
2157 rank = self._shape.ndims
2172 return (self._shape, self._dtype, self._ragged_rank, self._row_splits_dtype)
2177 return [tensor_spec.TensorSpec(self._shape, self._dtype)]
2180 self._shape[self._ragged_rank + 1:])
2181 outer_dim = tensor_shape.dimension_at_index(self._shape, 0)
2255 if self._shape.ndims is not None:
2257 outer_dim = tensor_shape.dimension_value(self._shape[0])
2262 self._shape[1 + self._ragged_rank:]))
2264 result.set_shape(self._shape)
[all …]
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_full_matrix.py176 def _shape(self): member in LinearOperatorFullMatrix
Dlinear_operator_inversion.py184 def _shape(self): member in LinearOperatorInversion
Dlinear_operator_lower_triangular.py177 def _shape(self): member in LinearOperatorLowerTriangular
Dlinear_operator_adjoint.py171 def _shape(self): member in LinearOperatorAdjoint
Dlinear_operator_diag.py176 def _shape(self): member in LinearOperatorDiag
Dlinear_operator_composition.py193 def _shape(self): member in LinearOperatorComposition
Dlinear_operator_identity.py294 def _shape(self): member in LinearOperatorIdentity
633 def _shape(self): member in LinearOperatorScaledIdentity
Dlinear_operator_householder.py166 def _shape(self): member in LinearOperatorHouseholder
Dlinear_operator_permutation.py182 def _shape(self): member in LinearOperatorPermutation
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc1149 auto _shape = tensorflow::ops::AsNodeOut(s, shape); in TEST_F() local
1152 builder.Input(_shape).Input(_value).Finalize(s.graph(), &func_op)); in TEST_F()
1175 auto _shape = tensorflow::ops::AsNodeOut(s, shape); in TEST_F() local
1178 builder.Input(_shape).Input(_value).Finalize(s.graph(), &func_op)); in TEST_F()
1205 auto _shape = tensorflow::ops::AsNodeOut(s, shape); in TEST_F() local
1209 TF_CHECK_OK(builder.Input(_shape).Finalize(s.graph(), &func_op)); in TEST_F()
1239 auto _shape = tensorflow::ops::AsNodeOut(s, shape); in TEST_F() local
1244 builder.Input(_shape).Input(_shape).Finalize(s.graph(), &func_op)); in TEST_F()

12