Searched refs:input_props (Results 1 – 5 of 5) sorted by relevance
71 const auto& input_props = properties.GetInputProperties(node.name()); in Optimize() local72 if (input_props.size() != 1) { in Optimize()76 const DataType type = input_props[0].dtype(); in Optimize()
681 const std::vector<OpInfo::TensorProperties>& input_props = in MaterializeReductionIndices() local683 if (input_props.size() != 2) { in MaterializeReductionIndices()686 const OpInfo::TensorProperties& input_prop = input_props[0]; in MaterializeReductionIndices()696 const OpInfo::TensorProperties& reduction_indices_prop = input_props[1]; in MaterializeReductionIndices()2483 const auto& input_props = properties.GetInputProperties(node.name())[0]; in IsReductionCandidateForSimplification() local2485 if (!input_props.has_shape() || input_props.shape().unknown_rank() || in IsReductionCandidateForSimplification()2489 *input_tensor_shape = input_props.shape(); in IsReductionCandidateForSimplification()2957 const auto& input_props = properties.GetInputProperties(conv_node->name()); in MulConvPushDown() local2958 if (input_props.size() < 2) { in MulConvPushDown()2961 const auto& filter_shape = input_props[1].shape(); in MulConvPushDown()
198 const std::vector<OpInfo::TensorProperties>& input_props = in IsGpuCompatible() local201 input_props.size() >= 2 ? input_props[1].shape() : TensorShapeProto(); in IsGpuCompatible()
382 OpInfo::TensorProperties input_props; in HasAllInputsBroadcastableToShape() local383 Status has_input_properties = GetTensorProperties(input, &input_props); in HasAllInputsBroadcastableToShape()385 ShapesBroadcastable(properties, input_props); in HasAllInputsBroadcastableToShape()1884 OpInfo::TensorProperties input_props; in ReshapeIsIdentity() local1887 !GetTensorProperties(reshape.input(0), &input_props).ok()) { in ReshapeIsIdentity()1891 return ShapesSymbolicallyEqual(input_props.shape(), reshape_props.shape()); in ReshapeIsIdentity()
867 const auto input_props = properties.GetInputProperties("init_restore"); in TEST_F() local868 ASSERT_EQ(2, input_props.size()); in TEST_F()869 const OpInfo::TensorProperties& input_prop = input_props[1]; in TEST_F()