Home
last modified time | relevance | path

Searched refs:extra_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dlinalg_ops_test.cc44 const string extra_shape = (op.name == "CholeskyGrad" ? ";?" : ""); in TEST() local
46 INFER_OK(op, "?" + extra_shape, "?"); in TEST()
48 "[1]" + extra_shape); in TEST()
50 "[1,2]" + extra_shape); in TEST()
52 INFER_OK(op, "[?,?]" + extra_shape, "[d0_0|d0_1,d0_0|d0_1]"); in TEST()
53 INFER_OK(op, "[1,?]" + extra_shape, "[d0_0,d0_0]"); in TEST()
54 INFER_OK(op, "[?,1]" + extra_shape, "[d0_1,d0_1]"); in TEST()
57 INFER_OK(op, "[5,?,7,?,?]" + extra_shape, in TEST()
59 INFER_OK(op, "[5,?,7,1,?]" + extra_shape, "[d0_0,d0_1,d0_2,d0_3,d0_3]"); in TEST()
60 INFER_OK(op, "[5,?,7,?,1]" + extra_shape, "[d0_0,d0_1,d0_2,d0_4,d0_4]"); in TEST()
Ddata_flow_ops.cc97 ShapeHandle extra_shape = c->UnknownShape(); in DynamicStitchShapeFunction() local
119 TF_RETURN_IF_ERROR(c->Merge(extra_shape, rest, &extra_shape)); in DynamicStitchShapeFunction()
135 TF_RETURN_IF_ERROR(c->Concatenate(output_shape, extra_shape, &output_shape)); in DynamicStitchShapeFunction()
/external/tensorflow/tensorflow/python/kernel_tests/
Dbatch_scatter_ops_test.py59 for extra_shape in (), (5,), (5, 9):
65 np.random.randn(*(indices_shape + extra_shape)), vtype)
67 old = _AsType(np.random.randn(*(indices_shape + extra_shape)), vtype)
Dscatter_ops_test.py139 for extra_shape in (), (5,), (5, 9):
159 np.random.randn(*(indices_shape + extra_shape)), vtype)
172 old = _AsType(np.random.randn(*((first_dim,) + extra_shape)), vtype)
Ddynamic_partition_op_test.py164 for extra_shape in (), (6,), (6, 7):
165 data = np.random.randn(*(shape + extra_shape))