Searched refs:n_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | data_flow_ops.cc | 29 Status DequeueManyV2Shape(InferenceContext* c, ShapeHandle n_shape) { in DequeueManyV2Shape() argument
35 c->Concatenate(n_shape, (*t)[i].shape, &combined_shape)); in DequeueManyV2Shape()
339 ShapeHandle n_shape; in __anona2f81a440602() local
341 n_shape = c->Vector(InferenceContext::kUnknownDim); in __anona2f81a440602()
347 n_shape = c->Vector(n); in __anona2f81a440602()
349 return DequeueManyV2Shape(c, n_shape); in __anona2f81a440602()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 1874 TF_ASSIGN_OR_RETURN(auto n_shape, builder.GetShape(n)); in Polygamma()
1876 if (n_shape != x_shape) { in Polygamma()
1880 n_shape.ToString(), x_shape.ToString()); in Polygamma()
1884 n_shape.element_type() == F16 || x_shape.element_type() == BF16; in Polygamma()
1890 XlaOp result = doit(n, x, n_shape.element_type()); in Polygamma()
1892 result = ConvertElementType(result, n_shape.element_type()); in Polygamma()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 588 n_shape = next_var.get_shape()
589 if not _ShapeLessThanOrEqual(n_shape, m_shape):
597 "less-specific shape." % (input_t.name, input_t.shape, n_shape))
|