Searched refs:input_properties (Results 1 – 5 of 5) sorted by relevance
523 absl::Span<const OpInfo::TensorProperties> input_properties = in OperationCanBeTranslatedToImplicitBatch() local525 if (absl::c_any_of(input_properties, [](const OpInfo::TensorProperties& p) { in OperationCanBeTranslatedToImplicitBatch()532 FindLeadingShape(input_properties); in OperationCanBeTranslatedToImplicitBatch()656 const std::vector<OpInfo::TensorProperties>& input_properties = in GetClusterBatchSizeForNode() local659 FindLeadingShape(GetInputsToDeterminateBatchSize(node, input_properties)); in GetClusterBatchSizeForNode()
316 const auto& input_properties = input_properties_map.at(node.name()); in VerboseLogUnknownDimensionSources() local320 for (const auto& input_prop : input_properties) { in VerboseLogUnknownDimensionSources()334 for (const auto& input_prop : input_properties) { in VerboseLogUnknownDimensionSources()2626 auto& input_properties = input_properties_[node.name()]; in InferStatically() local2629 CHECK_EQ(input_properties.size(), 0); in InferStatically()2631 input_properties.resize(ic->num_inputs()); in InferStatically()2635 &input_properties[i]); in InferStatically()2643 *input_properties[i].mutable_value() = raw_val; in InferStatically()2646 *input_properties[i].mutable_value() = *ctx->input_tensor_protos[i]; in InferStatically()2652 *input_properties[i].mutable_value() = MakeTensorProtoFromShape( in InferStatically()
598 auto& inputs = node_state.input_properties; in MaybeUpdateInputOutput()763 for (auto& input : node_state.input_properties) { in CreateOpContext()788 node_state.input_properties = in GetNodeStateOrCreateIt()
53 std::vector<OpInfo::TensorProperties> input_properties; member
204 for (const OpInfo::TensorProperties& input_properties : op_info.inputs()) { in CwiseOutputElementCount() local205 max_rank = std::max(max_rank, input_properties.shape().dim_size()); in CwiseOutputElementCount()219 for (const OpInfo::TensorProperties& input_properties : op_info.inputs()) { in CwiseOutputElementCount() local220 const TensorShapeProto& input_shape = input_properties.shape(); in CwiseOutputElementCount()