Searched refs:IndexIsValid (Results 1 – 6 of 6) sorted by relevance
33 TF_RET_CHECK(ShapeUtil::IndexIsValid(alias_.shape(), output_index)) in SetUpAlias()154 CHECK(ShapeUtil::IndexIsValid(alias_.shape(), output_index)) in GetAliasedParameter()197 TF_RET_CHECK(ShapeUtil::IndexIsValid(param_shape, alias.parameter_index)); in Verify()198 TF_RET_CHECK(ShapeUtil::IndexIsValid(output_shape, output_index)); in Verify()
118 TF_RET_CHECK(ShapeUtil::IndexIsValid( in Verify()121 TF_RET_CHECK(ShapeUtil::IndexIsValid( in Verify()
922 if (!ShapeUtil::IndexIsValid(*shape, index_)) {
500 static bool IndexIsValid(const Shape& shape, ShapeIndexView index);
769 /* static */ bool ShapeUtil::IndexIsValid(const Shape& shape, in IndexIsValid() function in xla::ShapeUtil
459 if (!ShapeUtil::IndexIsValid(parameter_shape, alias.param_index)) { in PopulateInputOutputAlias()