Home
last modified time | relevance | path

Searched refs:TF_RET_CHECK (Results 1 – 25 of 108) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config.cc30 TF_RET_CHECK(kind == AliasKind::kUserAlias || kind == AliasKind::kSystemAlias) in SetUpAlias()
32 TF_RET_CHECK(ShapeUtil::IndexIsValid(alias_.shape(), output_index)) in SetUpAlias()
36 TF_RET_CHECK(param_number >= 0) << param_number; in SetUpAlias()
37 TF_RET_CHECK(!OutputHasAlias(output_index)) in SetUpAlias()
40 TF_RET_CHECK(!alias_.element(output_index)) << absl::StrFormat( in SetUpAlias()
186 TF_RET_CHECK(0 <= alias.parameter_number); in Verify()
187 TF_RET_CHECK(entry->num_parameters() > alias.parameter_number); in Verify()
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()
198 TF_RET_CHECK(LayoutUtil::IsDenseArray(param_subshape)); in Verify()
[all …]
Dgeneric_transfer_manager.cc46 TF_RET_CHECK(elements.size() == ShapeUtil::TupleElementCount(shape)); in WriteSingleTupleIndexTable()
76 TF_RET_CHECK(executor->device_ordinal() == device_buffer.device_ordinal()); in TransferLiteralFromDeviceInternal()
80 TF_RET_CHECK(ShapeUtil::Equal(device_buffer.on_device_shape(), in TransferLiteralFromDeviceInternal()
110 TF_RET_CHECK(ShapeUtil::Equal(device_buffer.on_device_shape(), in TransferLiteralToDeviceAsync()
113 TF_RET_CHECK( in TransferLiteralToDeviceAsync()
115 TF_RET_CHECK(stream->parent()->device_ordinal() == in TransferLiteralToDeviceAsync()
125 TF_RET_CHECK(GetByteSizeRequirement(device_subshape) == in TransferLiteralToDeviceAsync()
Dhlo_schedule.cc45 TF_RET_CHECK(comp_it != id_to_computation.end()) in CreateFromProto()
58 TF_RET_CHECK(instr_it != id_to_instruction.end()) in CreateFromProto()
207 TF_RET_CHECK(sequences_.contains(computation->unique_id())) in Update()
243 TF_RET_CHECK(nonfusion_computations.size() == sequences_.size()) in Verify()
247 TF_RET_CHECK(sequences_.contains(computation->unique_id())) in Verify()
259 TF_RET_CHECK(instruction_position.insert({instruction, pos}).second) in Verify()
265 TF_RET_CHECK(instruction_position.size() == in Verify()
271 TF_RET_CHECK(instruction_position.contains(instruction)) in Verify()
277 TF_RET_CHECK(instruction_position.at(operand) < in Verify()
284 TF_RET_CHECK(instruction_position.at(pred) < in Verify()
Dhlo_verifier.cc480 TF_RET_CHECK(SameElementType(broadcast->shape(), operand_shape)); in HandleBroadcast()
481 TF_RET_CHECK(operand_shape.rank() == broadcast->dimensions().size()); in HandleBroadcast()
485 TF_RET_CHECK((output_dimension < broadcast->shape().rank()) && in HandleBroadcast()
497 TF_RET_CHECK(SameElementType(reshape->shape(), operand_shape)); in HandleReshape()
498 TF_RET_CHECK(ShapeUtil::ElementsIn(reshape->shape()) == in HandleReshape()
547 TF_RET_CHECK(custom_call != nullptr); in HandleCustomCall()
552 TF_RET_CHECK(LayoutUtil::HasLayout(custom_call->shape())); in HandleCustomCall()
553 TF_RET_CHECK(custom_call->operand_count() == in HandleCustomCall()
558 TF_RET_CHECK(ShapeUtil::Compatible(custom_call->operand(i)->shape(), in HandleCustomCall()
562 TF_RET_CHECK(LayoutUtil::HasLayout(operand_shape_with_layout)); in HandleCustomCall()
[all …]
Ddfs_hlo_visitor_with_default_test.cc43 TF_RET_CHECK(!(hlo->IsElementwise() && hlo->operand_count() == 2)) in TEST_F()
45 TF_RET_CHECK(!(hlo->IsElementwise() && hlo->operand_count() == 1)) in TEST_F()
52 TF_RET_CHECK(hlo->IsElementwise() && hlo->operand_count() == 2) in TEST_F()
58 TF_RET_CHECK(hlo->IsElementwise() && hlo->operand_count() == 1) in TEST_F()
Dlayout_assignment.cc186 TF_RET_CHECK(unconstrained_buffer_ids_.erase(buffer.id()) == 1) in SetBufferLayout()
255 TF_RET_CHECK(operand->shape().IsArray()); in SetArrayOperandLayout()
405 TF_RET_CHECK(data_shape.IsArray()); in BuildHostChannelConstraints()
406 TF_RET_CHECK(LayoutUtil::HasLayout(data_shape)); in BuildHostChannelConstraints()
409 TF_RET_CHECK(prev_layout == nullptr) in BuildHostChannelConstraints()
488 TF_RET_CHECK(send_buffer_shape.IsArray()); in AddMandatoryConstraints()
498 TF_RET_CHECK(recv_buffer_shape.IsArray()); in AddMandatoryConstraints()
519 TF_RET_CHECK(buffer_shape.IsArray()); in AddMandatoryConstraints()
540 TF_RET_CHECK(instruction->operand_count() == in AddMandatoryConstraints()
610 TF_RET_CHECK(instruction->branch_computation(j)->num_parameters() == 1); in AddMandatoryConstraints()
[all …]
Ddynamic_parameter_binding.cc27 TF_RET_CHECK(result.second); in Bind()
115 TF_RET_CHECK(dynamic_parameter.parameter_num >= 0 && in Verify()
117 TF_RET_CHECK(dynamic_dimension.parameter_num < entry->num_parameters()); in Verify()
118 TF_RET_CHECK(ShapeUtil::IndexIsValid( in Verify()
121 TF_RET_CHECK(ShapeUtil::IndexIsValid( in Verify()
124 TF_RET_CHECK( in Verify()
Dhlo_module.cc45 TF_RET_CHECK(schedule.module() == this); in set_schedule()
114 TF_RET_CHECK(it->get() == to_remove); in RemoveEmbeddedComputation()
251 TF_RET_CHECK(!ContainsKey(computation_names, computation->name())) in CheckUniqueNamesAndIdsForComputationsAndInstructions()
255 TF_RET_CHECK(!ContainsKey(computation_ids, computation->unique_id())) in CheckUniqueNamesAndIdsForComputationsAndInstructions()
260 TF_RET_CHECK(!ContainsKey(instruction_names, instruction->name())) in CheckUniqueNamesAndIdsForComputationsAndInstructions()
264 TF_RET_CHECK(!ContainsKey(instruction_ids, instruction->unique_id())) in CheckUniqueNamesAndIdsForComputationsAndInstructions()
280 TF_RET_CHECK(proto.has_host_program_shape()) in CreateFromProto()
283 TF_RET_CHECK(expected_program_shape.parameters_size() == in CreateFromProto()
288 TF_RET_CHECK(ShapeUtil::Compatible(expected_program_shape.parameters(i), in CreateFromProto()
298 TF_RET_CHECK( in CreateFromProto()
[all …]
Dhlo_module_group_metadata.cc98 TF_RET_CHECK(peer_tracked != nullptr) in Build()
100 TF_RET_CHECK(*tracked == *peer_tracked) in Build()
112 TF_RET_CHECK(companion_tracked != nullptr); in Build()
113 TF_RET_CHECK(*tracked == *companion_tracked); in Build()
161 TF_RET_CHECK(device) << "Instruction " << comm_instruction->ToString() in VerifyCompanionSets()
318 TF_RET_CHECK(channel_id_map_.find(*hlo->all_reduce_id()) == in RecordInstructions()
331 TF_RET_CHECK(all_reduce_map_.find(hlo->channel_id()) == in RecordInstructions()
346 TF_RET_CHECK(channel.send == nullptr) in RecordInstructions()
352 TF_RET_CHECK(channel.recv == nullptr) in RecordInstructions()
358 TF_RET_CHECK(channel.send_done == nullptr) in RecordInstructions()
[all …]
Dhlo_computation.cc223 TF_RET_CHECK(root_instruction() != instruction); in RemoveInstructionAndUnusedOperands()
225 TF_RET_CHECK(instruction->user_count() == 0); in RemoveInstructionAndUnusedOperands()
226 TF_RET_CHECK(IsRemovable(instruction)) in RemoveInstructionAndUnusedOperands()
253 TF_RET_CHECK(IsRemovable(instruction)) in RemoveInstruction()
255 TF_RET_CHECK(root_instruction() != instruction) in RemoveInstruction()
257 TF_RET_CHECK(instruction->user_count() == 0) in RemoveInstruction()
260 TF_RET_CHECK(instruction->control_predecessors().empty()) in RemoveInstruction()
263 TF_RET_CHECK(instruction->control_successors().empty()) in RemoveInstruction()
268 TF_RET_CHECK(inst_it != instruction_iterators_.end()); in RemoveInstruction()
550 TF_RET_CHECK(!ContainsKey(instruction_map, instruction_proto.id())); in CreateFromProto()
[all …]
Dcall_inliner.cc44 TF_RET_CHECK(hlo->opcode() != HloOpcode::kCall); in DefaultAction()
113 TF_RET_CHECK(result.second) in NoteMapping()
127 TF_RET_CHECK(call->opcode() == HloOpcode::kCall) in Inline()
130 TF_RET_CHECK(callees.size() == 1); in Inline()
Dhlo_sharding_metadata.cc188 TF_RET_CHECK(!lhs->IsTuple() && !rhs.IsTuple()); in AssignLeafSharding()
214 TF_RET_CHECK(lhs_tree->IsLeaf(lhs_it->first)); in AssignTreeSharding()
227 TF_RET_CHECK(rhs_it == rhs_tree.end()); in AssignTreeSharding()
237 TF_RET_CHECK(instruction->shape().IsTuple() && in ApplyShardingFromUsers()
284 TF_RET_CHECK(user_sharding_tree.leaf_count() == 1) in ApplyShardingFromUsers()
304 TF_RET_CHECK(sharding_tree.leaf_count() == 1); in ApplyShardingFromUsers()
364 TF_RET_CHECK(instruction->shape().IsTuple()) in ApplyDomainSharding()
387 TF_RET_CHECK(ShardingMatches(*sharding, instruction->sharding())) in ExtractOriginalCommonSharding()
Dhlo_instruction.cc66 TF_RET_CHECK(!proto.opcode().empty()); in CreateFromProto()
96 TF_RET_CHECK(proto.has_shape()); in CreateFromProto()
123 TF_RET_CHECK( in CreateFromProto()
128 TF_RET_CHECK( in CreateFromProto()
138 TF_RET_CHECK(proto.operand_ids_size() == *arity) in CreateFromProto()
208 TF_RET_CHECK(proto.dimensions_size() == 1) in CreateFromProto()
215 TF_RET_CHECK(proto.called_computation_ids_size() > 0) in CreateFromProto()
218 TF_RET_CHECK(proto.called_computation_ids_size() == 2) in CreateFromProto()
222 TF_RET_CHECK(proto.operand_ids_size() == in CreateFromProto()
235 TF_RET_CHECK(proto.operand_ids_size() % 2 == 0) in CreateFromProto()
[all …]
Dcopy_insertion.cc152 TF_RET_CHECK(to_copy == nullptr); in DeepCopyAndAddControlEdges()
155 TF_RET_CHECK(to_copy != nullptr); in DeepCopyAndAddControlEdges()
259 TF_RET_CHECK(xla_while->opcode() == HloOpcode::kWhile); in AddCopiesForWhile()
291 TF_RET_CHECK(param != root); in AddCopiesForWhile()
321 TF_RET_CHECK(conditional->opcode() == HloOpcode::kConditional); in AddCopiesForConditional()
367 TF_RET_CHECK(param->parameter_number() < entry->num_parameters()); in AddCopiesForAliasedInputOutputs()
368 TF_RET_CHECK(!copied_parameters[param->parameter_number()]); in AddCopiesForAliasedInputOutputs()
410 TF_RET_CHECK(from != nullptr); in AddCopiesForAliasedInputOutputs()
411 TF_RET_CHECK(to != nullptr); in AddCopiesForAliasedInputOutputs()
600 TF_RET_CHECK(p->prev->next == p); in Verify()
[all …]
Dcomputation_placer.cc56 TF_RET_CHECK(proto.computation_devices_size() == proto.computation_count()); in Deserialize()
68 TF_RET_CHECK(computation_device.replica_device_ids_size() == in Deserialize()
94 TF_RET_CHECK(replica < replica_count); in DeviceId()
95 TF_RET_CHECK(computation < computation_count); in DeviceId()
Dhlo_domain_verifier.cc52 TF_RET_CHECK(instruction->user_side_metadata().Kind() == in PopulateDomainKinds()
104 TF_RET_CHECK(meta.Matches(*ref_metadata)) in VerifyDomain()
116 TF_RET_CHECK(meta.Matches(*ref_metadata)) in VerifyDomain()
Dhlo_module_group.cc71 TF_RET_CHECK(!proto.name().empty()) << "Module group name cannot be empty"; in CreateFromProto()
72 TF_RET_CHECK(proto.hlo_modules_size() > 0) in CreateFromProto()
74 TF_RET_CHECK(proto.hlo_modules_size() == module_configs.size()); in CreateFromProto()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.cc33 TF_RET_CHECK(buffer.builder()); in BuildTensorList()
39 TF_RET_CHECK(op.builder()); in GetTensorListBuffer()
45 TF_RET_CHECK(op.builder()); in GetTensorListPushIndex()
52 TF_RET_CHECK(op.builder()); in GetTensorListBufferShape()
61 TF_RET_CHECK(list_shape.IsTuple()); in GetTensorListBufferShape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinfeed_thunk.cc41 TF_RET_CHECK(infeed_shape.IsTuple()) in ExecuteOnStream()
43 TF_RET_CHECK(infeed_shape.tuple_shapes().size() == 2) in ExecuteOnStream()
45 TF_RET_CHECK(infeed_shape.tuple_shapes(1).IsToken()) in ExecuteOnStream()
47 TF_RET_CHECK( in ExecuteOnStream()
Dir_emitter.cc154 TF_RET_CHECK(nested_computation.num_parameters() > 0); in EmitCallToNestedComputation()
420 TF_RET_CHECK(pred->shape().element_type() == PRED); in HandleSelect()
431 TF_RET_CHECK(pred->shape().element_type() == PRED); in HandleTupleSelect()
432 TF_RET_CHECK(ShapeUtil::IsScalar(pred->shape())); in HandleTupleSelect()
433 TF_RET_CHECK(tuple_select->shape().IsTuple()); in HandleTupleSelect()
498 TF_RET_CHECK(ShapeUtil::ElementIsIntegral(lhs_shape)); in HandleDot()
507 TF_RET_CHECK(!ShapeUtil::IsScalar(lhs_shape) && in HandleDot()
520 TF_RET_CHECK(lhs_shape.dimensions(lhs_reduction_dimension) == in HandleDot()
576 TF_RET_CHECK(!reduction_loop->GetBodyBasicBlock()->empty()); in HandleDot()
599 TF_RET_CHECK(ShapeUtil::ElementIsIntegral(lhs_shape)); in HandleDot()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc83 TF_RET_CHECK(TensorShapeUtils::IsVector(arg.shape())); in GetComputationInputs()
146 TF_RET_CHECK(TensorShapeUtils::IsScalar(execution_input.shape())); in DoWork()
150 TF_RET_CHECK(TensorShapeUtils::IsScalar(execution_config.shape())); in DoWork()
152 TF_RET_CHECK( in DoWork()
156 TF_RET_CHECK(core_index_in_replica == 0); in DoWork()
241 TF_RET_CHECK(alias.parameter_number < input_tuples.size()); in DoWork()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing.cc25 TF_RET_CHECK(start.size() == end.size()); in SliceInMinorDims()
31 TF_RET_CHECK(n_minor_dims <= n_dims); in SliceInMinorDims()
57 TF_RET_CHECK(start.size() == n_dims); in UpdateSlice()
76 TF_RET_CHECK(n_minor_dims <= n_dims); in UpdateSliceInMinorDims()
116 TF_RET_CHECK(n_minor_dims == sizes.size()); in DynamicSliceInMinorDims()
117 TF_RET_CHECK(n_minor_dims <= n_dims); in DynamicSliceInMinorDims()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.cc190 TF_RET_CHECK(ShapeUtil::Compatible(received_shape, literal.shape())) in TransferLiteralFromOutfeed()
195 TF_RET_CHECK(size == GetByteSizeRequirement(received_shape)); in TransferLiteralFromOutfeed()
216 TF_RET_CHECK(ShapeUtil::Compatible(received_shape, literal_shape)) in TransferLiteralFromOutfeed()
221 TF_RET_CHECK(GetByteSizeRequirement(literal_shape) == in TransferLiteralFromOutfeed()
224 TF_RET_CHECK(ShapeUtil::Equal(literal.shape(), literal_shape)); in TransferLiteralFromOutfeed()
282 TF_RET_CHECK(outfed_shapes.size() == 1); in TransferBuffersFromOutfeedInternal()
/external/tensorflow/tensorflow/compiler/xla/
Dstatus_macros_test.cc26 TF_RET_CHECK(2 > 3); in RetCheckFail()
31 TF_RET_CHECK(2 > 3) << "extra message"; in RetCheckFailWithExtraMessage()
36 TF_RET_CHECK(3 > 2); in RetCheckSuccess()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.cc375 TF_RET_CHECK(variable != nullptr); in ReadVariableInputTensor()
376 TF_RET_CHECK(variable->kind() == XlaResource::kVariable); in ReadVariableInputTensor()
425 TF_RET_CHECK(variable != nullptr); in GetVariableTypeAndShape()
426 TF_RET_CHECK(variable->kind() == XlaResource::kVariable); in GetVariableTypeAndShape()
503 TF_RET_CHECK(expression->resource() != nullptr); in GetResourceInput()
515 TF_RET_CHECK(variable != nullptr); in AssignVariableTensor()
516 TF_RET_CHECK(variable->kind() == XlaResource::kVariable); in AssignVariableTensor()
545 TF_RET_CHECK(handle.valid()); in AssignVariable()
552 TF_RET_CHECK(handle.valid()); in AssignVariable()

12345