Searched refs:IndexIsValid (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_input_output_alias_config.cc | 32 TF_RET_CHECK(ShapeUtil::IndexIsValid(alias_.shape(), output_index)) in SetUpAlias()
153 CHECK(ShapeUtil::IndexIsValid(alias_.shape(), output_index)) in GetAliasedParameter()
196 TF_RET_CHECK(ShapeUtil::IndexIsValid(param_shape, alias.parameter_index)); in Verify()
197 TF_RET_CHECK(ShapeUtil::IndexIsValid(output_shape, output_index)); in Verify()
|
| D | dynamic_parameter_binding.cc | 118 TF_RET_CHECK(ShapeUtil::IndexIsValid( in Verify()
121 TF_RET_CHECK(ShapeUtil::IndexIsValid( in Verify()
|
| D | hlo_verifier.cc | 816 TF_RET_CHECK(ShapeUtil::IndexIsValid( in HandleCustomCall()
819 TF_RET_CHECK(ShapeUtil::IndexIsValid(custom_call->shape(), pair.first)) in HandleCustomCall()
|
| D | dynamic_dimension_inference.cc | 1733 CHECK(ShapeUtil::IndexIsValid(inst->shape(), index)); in GetDynamicSizes()
|
| D | pattern_matcher.h | 909 if (!ShapeUtil::IndexIsValid(*shape, index_)) {
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.cc | 107 if (!ShapeUtil::IndexIsValid(alias_config.shape(), result_index)) { in AllocateOutputMemoryWithInputReuse()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | utils.cc | 175 CHECK(ShapeUtil::IndexIsValid(sharded_shape, idx)); in DetermineArgumentLayoutsFromCompileOptions()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 186 equal &= IndexIsValid(rhs, index); in EqualStructure()
189 equal &= IndexIsValid(lhs, index); in EqualStructure()
896 /* static */ bool ShapeUtil::IndexIsValid(const Shape& shape, in IndexIsValid() function in xla::ShapeUtil
|
| D | shape_util.h | 549 static bool IndexIsValid(const Shape& shape, ShapeIndexView index);
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 538 if (!ShapeUtil::IndexIsValid(parameter_shape, alias.param_index)) { in PopulateInputOutputAlias()
|