HomeSort by relevance Sort by last modified time
    Searched refs:dims_actual (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/framework/
common_shape_fns.cc 299 std::vector<DimensionHandle> dims_actual(num_dims);
300 dims_actual[GetTensorBatchDimIndex(num_dims, format)] = context->MakeDim(N);
302 dims_actual[outer_c_index] = context->MakeDim(C);
304 dims_actual[GetTensorInnerFeatureDimIndex(num_dims, format)] =
308 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] =
311 *out = context->MakeShape(dims_actual);
    [all...]
shape_inference.cc 682 std::vector<DimensionHandle> dims_actual; local
683 dims_actual.reserve(dims.size());
685 dims_actual.push_back(MakeDim(d));
688 return shape_manager_.MakeShape(dims_actual);
    [all...]

Completed in 538 milliseconds