Searched refs:data_shape_tf (Results 1 – 1 of 1) sorted by relevance
609 TensorShape data_shape_tf = ctx->InputShape("data"); in Compile() local610 OP_REQUIRES(ctx, data_shape_tf.dims() == 4, in Compile()612 data_shape_tf.DebugString())); in Compile()613 const int64 data_channels = data_shape_tf.dim_size(3); in Compile()640 xla::Shape data_shape = TensorShapeToXLAShape(data_type, data_shape_tf); in Compile()