Searched refs:paddings_tensor (Results 1 – 4 of 4) sorted by relevance
30 Tensor paddings_tensor(DT_INT32, TensorShape({num_block_dims, 2})); in ConstructSpaceToBatchGraph() local31 auto paddings_eigen_tensor = paddings_tensor.matrix<int32>(); in ConstructSpaceToBatchGraph()42 .Input(test::graph::Constant(g, paddings_tensor)) in ConstructSpaceToBatchGraph()50 .Input(test::graph::Constant(g, paddings_tensor)) in ConstructSpaceToBatchGraph()
95 const int64 paddings_tensor[NUM_BLOCK_DIMS * 2], in operator ()()108 pad_start[block_dim] = paddings_tensor[block_dim * 2]; in operator ()()
166 Tensor paddings_tensor(DT_INT32, paddings_shape); in BM_ConvFloat() local167 for (int64 i = 0; i < paddings_tensor.NumElements(); ++i) { in BM_ConvFloat()168 paddings_tensor.flat<int32>()(i) = 0; in BM_ConvFloat()172 .Attr("value", paddings_tensor) in BM_ConvFloat()
102 Tensor paddings_tensor(DT_INT32, paddings_shape); in TEST() local103 for (int64 i = 0; i < paddings_tensor.NumElements(); ++i) { in TEST()104 paddings_tensor.flat<int32>()(i) = 0; in TEST()108 .Attr("value", paddings_tensor) in TEST()