Searched refs:out_dim (Results 1 – 5 of 5) sorted by relevance
155 DimensionHandle out_dim = c.Dim(out, j); in InferShapes() local161 if (c.Dim(in, j).SameHandle(out_dim)) { in InferShapes()175 } else if (c.ValueKnown(out_dim)) { in InferShapes()177 c.Value(out_dim), err_suffix); in InferShapes()218 } else if (value != c.Value(out_dim)) { in InferShapes()220 c.DebugString(out_dim), err_suffix); in InferShapes()
1051 for (int64 out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims() local1052 const int64 in_dim = out_dim + offset; in ComputeFlatInnerDims()1053 out_dims[out_dim] = in_dim < 0 ? 1 : orig[in_dim]; in ComputeFlatInnerDims()1064 for (int64 out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims() local1065 out_dims[out_dim] = out_dim >= orig.size() ? 1 : orig[out_dim]; in ComputeFlatOuterDims()
39 auto* out_dim = out->add_dim(); in ProtoFromShapeHandle() local41 out_dim->set_size(c->Value(d)); in ProtoFromShapeHandle()43 out_dim->set_size(-1); in ProtoFromShapeHandle()
1305 DSizes<ptrdiff_t, 3> out_dim; in TEST() local1306 out_dim[0] = 1; in TEST()1307 out_dim[1] = 4; in TEST()1308 out_dim[2] = 12; in TEST()1346 .reshape(out_dim), in TEST()1357 .reshape(out_dim), in TEST()
298 for target_dim, out_dim in zip(y.shape[1:], shape[1:]):299 if out_dim is not None and target_dim != out_dim: