Searched refs:op_child_input_props (Results 1 – 2 of 2) sorted by relevance
3159 ctx->op_child_input_props = in PrepareConstantPushDown()3163 ctx->op_child_input_props == nullptr || in PrepareConstantPushDown()3164 ctx->op_child_input_props->size() < 2)) { in PrepareConstantPushDown()3224 if (ctx.parent_input_props->empty() || ctx.op_child_input_props->empty() || in ConstantPushDownBiasAdd()3227 (*ctx.op_child_input_props)[0].shape().unknown_rank() || in ConstantPushDownBiasAdd()3228 (*ctx.op_child_input_props)[1].shape().unknown_rank()) { in ConstantPushDownBiasAdd()3233 const int left_leaf_rank = (*ctx.op_child_input_props)[0].shape().dim_size(); in ConstantPushDownBiasAdd()3234 const int right_leaf_rank = (*ctx.op_child_input_props)[1].shape().dim_size(); in ConstantPushDownBiasAdd()3240 const auto& vector_prop = (*ctx.op_child_input_props)[vector_idx]; in ConstantPushDownBiasAdd()3245 const auto& matrix_prop = (*ctx.op_child_input_props)[matrix_idx]; in ConstantPushDownBiasAdd()[all …]
172 const std::vector<OpInfo::TensorProperties>* op_child_input_props; member