Searched refs:val_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_state.py | 607 val_shape = val.get_shape()
617 if val_shape.is_fully_defined():
621 result = array_ops.zeros(val_shape.dims, val.dtype)
639 if val_shape.is_fully_defined():
642 result = array_ops.zeros(val_shape.dims, val.dtype)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_concat_op_test.py | 36 val_shape=None, argument
42 dtypes.float32, shape=val_shape),
333 self._SparseTensor_UnknownShape(val_shape=[3]),
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 332 const Shape& val_shape,
|
| D | shape_inference.cc | 2323 const Shape& shape, const Shape& val_shape, int64 dimension) { in InferSetDimensionSizeShape() argument
2329 if (val_shape.rank() != 0 || val_shape.element_type() != S32) { in InferSetDimensionSizeShape()
2332 val_shape.ToString()); in InferSetDimensionSizeShape()
|
| D | shape_inference_test.cc | 1629 Shape val_shape = ShapeUtil::MakeShape(S32, {1}); in TEST_F() local
1631 arg_shape, val_shape, /*dimension=*/0); in TEST_F()
1640 Shape val_shape = ShapeUtil::MakeShape(U32, {}); in TEST_F() local
1642 arg_shape, val_shape, /*dimension=*/0); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 3366 TF_ASSIGN_OR_RETURN(const Shape* val_shape, GetShapePtr(val)); in SetDimensionSize()
3370 *operand_shape, *val_shape, dimension)); in SetDimensionSize()
|