Home
last modified time | relevance | path

Searched refs:op_child_input_props (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc3159 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 …]
Dconstant_folding.h172 const std::vector<OpInfo::TensorProperties>* op_child_input_props; member