HomeSort by relevance Sort by last modified time
    Searched full:output_parameters_shape (Results 1 - 1 of 1) sorted by null

  /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
stochastic_hard_routing_gradient_op.cc 143 TensorShape output_parameters_shape; variable
144 output_parameters_shape.AddDim(num_data);
145 output_parameters_shape.AddDim(num_nodes);
146 output_parameters_shape.AddDim(num_features);
156 OP_REQUIRES_OK(context, context->allocate_output(2, output_parameters_shape,

Completed in 83 milliseconds