Searched refs:IndexIsValid (Results 1 – 6 of 6) sorted by relevance
32 TF_RET_CHECK(ShapeUtil::IndexIsValid(alias_.shape(), output_index)) in SetUpAlias()149 CHECK(ShapeUtil::IndexIsValid(alias_.shape(), output_index)); in GetAliasedParameter()191 TF_RET_CHECK(ShapeUtil::IndexIsValid(param_shape, alias.parameter_index)); in Verify()192 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()
928 if (!ShapeUtil::IndexIsValid(*shape, index_)) {
487 static bool IndexIsValid(const Shape& shape, ShapeIndexView index);
769 /* static */ bool ShapeUtil::IndexIsValid(const Shape& shape, in IndexIsValid() function in xla::ShapeUtil
405 if (!ShapeUtil::IndexIsValid(parameter_shape, alias.param_index)) { in PopulateInputOutputAlias()