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()
740 const std::vector<OpInfo::TensorProperties>& input_props = in MaterializeReductionIndices() local742 if (input_props.size() != 2) { in MaterializeReductionIndices()745 const OpInfo::TensorProperties& input_prop = input_props[0]; in MaterializeReductionIndices()755 const OpInfo::TensorProperties& reduction_indices_prop = input_props[1]; in MaterializeReductionIndices()1006 const std::vector<OpInfo::TensorProperties>& input_props = in IsFoldableUncached() local1012 for (const auto& input_prop : input_props) { in IsFoldableUncached()2564 const std::vector<OpInfo::TensorProperties>& input_props = in SimplifySelect() local2566 if (input_props.size() < 3) return false; in SimplifySelect()2575 const TensorShapeProto& predicate_shape = input_props[0].shape(); in SimplifySelect()2578 if (ShapesSymbolicallyEqual(input_props[1], input_props[2]) && in SimplifySelect()[all …]
469 const OpInfo::TensorProperties* input_props; in HasAllInputsBroadcastableToShape() local470 Status has_input_properties = GetTensorProperties(input, &input_props); in HasAllInputsBroadcastableToShape()472 ShapesBroadcastable(properties, *input_props); in HasAllInputsBroadcastableToShape()2073 const OpInfo::TensorProperties* input_props; in InputMatchesTargetShape() local2075 !GetTensorProperties(reshape.input(0), &input_props).ok()) { in InputMatchesTargetShape()2079 return ShapesSymbolicallyEqual(input_props->shape(), in InputMatchesTargetShape()2946 const OpInfo::TensorProperties* input_props; in TrySimplify() local2947 TF_RETURN_IF_ERROR(GetTensorProperties(input->name(), &input_props)); in TrySimplify()2948 const TensorShapeProto& input_shape = input_props->shape(); in TrySimplify()
474 const std::vector<OpInfo::TensorProperties>& input_props = in IsGpuCompatible() local477 input_props.size() >= 2 ? input_props[1].shape() : TensorShapeProto(); in IsGpuCompatible()1568 auto input_props = in FindMklLayerNorm() local1574 if (ShapesSymbolicallyEqual(input_props[0].shape(), in FindMklLayerNorm()1576 int rank = Rank(input_props[0].shape()); in FindMklLayerNorm()
1073 const auto input_props = properties.GetInputProperties("init_restore"); in TEST_F() local1074 ASSERT_EQ(2, input_props.size()); in TEST_F()1075 const OpInfo::TensorProperties& input_prop = input_props[1]; in TEST_F()