Searched refs:input_props (Results 1 – 5 of 5) sorted by relevance
109 const auto& input_props = properties.GetInputProperties(node.name()); in Optimize() local110 if (input_props.size() != 1) { in Optimize()115 const DataType type = input_props[0].dtype(); in Optimize()
736 const std::vector<OpInfo::TensorProperties>& input_props = in MaterializeReductionIndices() local738 if (input_props.size() != 2) { in MaterializeReductionIndices()741 const OpInfo::TensorProperties& input_prop = input_props[0]; in MaterializeReductionIndices()751 const OpInfo::TensorProperties& reduction_indices_prop = input_props[1]; in MaterializeReductionIndices()1002 const std::vector<OpInfo::TensorProperties>& input_props = in IsFoldableUncached() local1008 for (const auto& input_prop : input_props) { in IsFoldableUncached()2514 const std::vector<OpInfo::TensorProperties>& input_props = in SimplifySelect() local2516 if (input_props.size() < 3) return false; in SimplifySelect()2525 const TensorShapeProto& predicate_shape = input_props[0].shape(); in SimplifySelect()2528 if (ShapesSymbolicallyEqual(input_props[1], input_props[2]) && in SimplifySelect()[all …]
335 const std::vector<OpInfo::TensorProperties>& input_props = in IsCpuCompatibleDataType() local338 input_props.size() >= 2 ? input_props[1].shape() : TensorShapeProto(); in IsCpuCompatibleDataType()
407 const OpInfo::TensorProperties* input_props; in HasAllInputsBroadcastableToShape() local408 Status has_input_properties = GetTensorProperties(input, &input_props); in HasAllInputsBroadcastableToShape()410 ShapesBroadcastable(properties, *input_props); in HasAllInputsBroadcastableToShape()1985 const OpInfo::TensorProperties* input_props; in InputMatchesTargetShape() local1988 !GetTensorProperties(reshape.input(0), &input_props).ok()) { in InputMatchesTargetShape()1992 return ShapesSymbolicallyEqual(input_props->shape(), in InputMatchesTargetShape()
905 const auto input_props = properties.GetInputProperties("init_restore"); in TEST_F() local906 ASSERT_EQ(2, input_props.size()); in TEST_F()907 const OpInfo::TensorProperties& input_prop = input_props[1]; in TEST_F()