Searched defs:in_shape (Results 1 – 12 of 12) sorted by relevance
43 const TensorShape in_shape = ctx->InputShape(i); in Compile() local92 const TensorShape in_shape = ctx->InputShape(i); in Compile() local
37 const TensorShape in_shape = ctx->InputShape(0); in Compile() local
80 const TensorShape& in_shape = shapes[i]; in Compile() local
32 TensorShape in_shape(chunk_size); in MakeGraph() local
60 TensorShape in_shape(total_size); in MakeGraph() local
40 const auto in_shape = tensor_in.shape(); in Compute() local
260 TensorShape in_shape; in Compute() local
336 TensorShape in_shape = context->input(0).shape(); in HandleCase() local
245 Shape in_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {0, 1}); in XLA_TEST_F() local
150 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
1882 auto in_shape = make_shape(in_dims, options.input_minor_to_major_layout); in TEST_F() local
1552 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local1626 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local1732 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local